add asssert msg for local data env

agimus-project · Oct 25, 2024 · 66ebb7b · 66ebb7b
1 parent 580b1cd
commit 66ebb7b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/happypose/pose_estimators/cosypose/cosypose/config.py b/happypose/pose_estimators/cosypose/cosypose/config.py
@@ -17,9 +17,11 @@
 PROJECT_ROOT = Path(happypose.__file__).parent.parent
 PROJECT_DIR = PROJECT_ROOT
 DATA_DIR = PROJECT_DIR / "data"
+# user should set "HAPPYPOSE_DATA_DIR" env or create project dir "local_data"
 LOCAL_DATA_DIR = Path(
     os.environ.get("HAPPYPOSE_DATA_DIR", Path(PROJECT_DIR) / "local_data"),
 )
+assert LOCAL_DATA_DIR.exists(), "Did you forget to set env variable 'HAPPYPOSE_DATA_DIR'?"
 TEST_DATA_DIR = LOCAL_DATA_DIR
 DASK_LOGS_DIR = LOCAL_DATA_DIR / "dasklogs"
 SYNT_DS_DIR = LOCAL_DATA_DIR / "synt_datasets"
@@ -36,7 +38,6 @@
 DEPS_DIR = PROJECT_DIR / "deps"
 CACHE_DIR = LOCAL_DATA_DIR / "joblib_cache"
 
-assert LOCAL_DATA_DIR.exists()
 CACHE_DIR.mkdir(exist_ok=True)
 TEST_DATA_DIR.mkdir(exist_ok=True)
 DASK_LOGS_DIR.mkdir(exist_ok=True)

diff --git a/happypose/pose_estimators/megapose/config.py b/happypose/pose_estimators/megapose/config.py
@@ -27,9 +27,11 @@
 
 PROJECT_ROOT = Path(happypose.__file__).parent.parent
 PROJECT_DIR = PROJECT_ROOT
+# user should set "HAPPYPOSE_DATA_DIR" env or create project dir "local_data"
 LOCAL_DATA_DIR = Path(
     os.environ.get("HAPPYPOSE_DATA_DIR", Path(PROJECT_DIR) / "local_data"),
 )
+assert LOCAL_DATA_DIR.exists(), "Did you forget to set env variable 'HAPPYPOSE_DATA_DIR'?"
 BOP_DS_DIR = LOCAL_DATA_DIR / "bop_datasets"
 NB_DATA_DIR = LOCAL_DATA_DIR / "notebook_data"
 SHAPENET_DIR = LOCAL_DATA_DIR / "shapenetcorev2"