Skip to content

Commit 57bd589

Browse files
committed
path agnosticity
1 parent 25e3b82 commit 57bd589

File tree

1 file changed

+26
-14
lines changed

1 file changed

+26
-14
lines changed

src/analysis/analysis.rs

+26-14
Original file line numberDiff line numberDiff line change
@@ -259,33 +259,45 @@ impl Analysis {
259259
"#).await?)
260260
}
261261
async fn copy_blueprint(&self) -> Result<(), E> {
262-
Ok(self.0.batch_execute(r#"
263-
COPY blueprint (past, present, future, edge, policy, regret) FROM '/Users/krukah/Code/robopoker/blueprint.profile.pgcopy' WITH (FORMAT BINARY);
262+
let path = std::env::current_dir()
263+
.unwrap()
264+
.to_string_lossy()
265+
.into_owned();
266+
Ok(self.0.batch_execute(format!(r#"
267+
COPY blueprint (past, present, future, edge, policy, regret) FROM '{}/blueprint.profile.pgcopy' WITH (FORMAT BINARY);
264268
CREATE INDEX IF NOT EXISTS idx_blueprint_bucket ON blueprint (present, past, future);
265269
CREATE INDEX IF NOT EXISTS idx_blueprint_future ON blueprint (future);
266270
CREATE INDEX IF NOT EXISTS idx_blueprint_present ON blueprint (present);
267271
CREATE INDEX IF NOT EXISTS idx_blueprint_edge ON blueprint (edge);
268272
CREATE INDEX IF NOT EXISTS idx_blueprint_past ON blueprint (past);
269-
"#).await?)
273+
"#, path).as_str()).await?)
270274
}
271275
async fn copy_metric(&self) -> Result<(), E> {
272-
Ok(self.0.batch_execute(r#"
273-
COPY metric (xor, dx) FROM '/Users/krukah/Code/robopoker/turn.metric.pgcopy' WITH (FORMAT BINARY);
274-
COPY metric (xor, dx) FROM '/Users/krukah/Code/robopoker/flop.metric.pgcopy' WITH (FORMAT BINARY);
275-
COPY metric (xor, dx) FROM '/Users/krukah/Code/robopoker/preflop.metric.pgcopy' WITH (FORMAT BINARY);
276+
let path = std::env::current_dir()
277+
.unwrap()
278+
.to_string_lossy()
279+
.into_owned();
280+
Ok(self.0.batch_execute(format!(r#"
281+
COPY metric (xor, dx) FROM '{}/turn.metric.pgcopy' WITH (FORMAT BINARY);
282+
COPY metric (xor, dx) FROM '{}/flop.metric.pgcopy' WITH (FORMAT BINARY);
283+
COPY metric (xor, dx) FROM '{}/preflop.metric.pgcopy' WITH (FORMAT BINARY);
276284
CREATE INDEX IF NOT EXISTS idx_metric_xor ON metric (xor);
277285
CREATE INDEX IF NOT EXISTS idx_metric_dx ON metric (dx);
278-
"#).await?)
286+
"#, path, path, path).as_str()).await?)
279287
}
280288
async fn copy_encoder(&self) -> Result<(), E> {
281-
Ok(self.0.batch_execute(r#"
282-
COPY encoder (obs, abs) FROM '/Users/krukah/Code/robopoker/river.encoder.pgcopy' WITH (FORMAT BINARY);
283-
COPY encoder (obs, abs) FROM '/Users/krukah/Code/robopoker/turn.encoder.pgcopy' WITH (FORMAT BINARY);
284-
COPY encoder (obs, abs) FROM '/Users/krukah/Code/robopoker/flop.encoder.pgcopy' WITH (FORMAT BINARY);
285-
COPY encoder (obs, abs) FROM '/Users/krukah/Code/robopoker/preflop.encoder.pgcopy' WITH (FORMAT BINARY);
289+
let path = std::env::current_dir()
290+
.unwrap()
291+
.to_string_lossy()
292+
.into_owned();
293+
Ok(self.0.batch_execute(format!(r#"
294+
COPY encoder (obs, abs) FROM '{}/river.encoder.pgcopy' WITH (FORMAT BINARY);
295+
COPY encoder (obs, abs) FROM '{}/turn.encoder.pgcopy' WITH (FORMAT BINARY);
296+
COPY encoder (obs, abs) FROM '{}/flop.encoder.pgcopy' WITH (FORMAT BINARY);
297+
COPY encoder (obs, abs) FROM '{}/preflop.encoder.pgcopy' WITH (FORMAT BINARY);
286298
CREATE INDEX IF NOT EXISTS idx_encoder_obs ON encoder (obs);
287299
CREATE INDEX IF NOT EXISTS idx_encoder_abs ON encoder (abs);
288-
"#).await?)
300+
"#, path, path, path, path).as_str()).await?)
289301
}
290302
async fn copy_abstraction(&self) -> Result<(), E> {
291303
Ok(self.0.batch_execute(r#"

0 commit comments

Comments
 (0)