Skip to content

Commit c41b793

Browse files
author
Frankie Robertson
committed
Add SETUP_WORK to Dockerfile env to allow easily getting GCN_CONFIG from container
1 parent e4df212 commit c41b793

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

docker/skelshop_env

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@ export LC_ALL=C.UTF-8
33
export LANG=C.UTF-8
44
export LD_LIBRARY_PATH=$OPENPOSE/src/openpose/:$LD_LIBRARY_PATH
55
export MODEL_FOLDER=$OPENPOSE_MODELS
6+
export SETUP_WORK=/opt/skelshop/work/

workflow/Snakefile

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,11 @@ def cnf(name, val):
99

1010
# Intermediate dirs
1111
cnf("WORK", "work")
12+
SETUP_WORK = os.environ.get("SETUP_WORK", WORK)
1213
cnf("VIDEO_BASE", ".")
1314
cnf("DUMP_BASE", ".")
14-
cnf("GCN_WEIGHTS", pjoin(WORK, "gcn_weights"))
15-
cnf("GCN_CONFIG", pjoin(WORK, "gcn_config.yaml"))
15+
cnf("GCN_WEIGHTS", pjoin(SETUP_WORK, "gcn_weights"))
16+
cnf("GCN_CONFIG", pjoin(SETUP_WORK, "gcn_config.yaml"))
1617
cnf("FFPROBE_BIN", which("ffprobe"))
1718

1819
## Configs

0 commit comments

Comments
 (0)