Ensure embedding cache dir exists in examples and fix typo

computational-cell-analytics · Nov 18, 2023 · d8457fc · d8457fc
1 parent 1c79323
commit d8457fc
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/examples/annotator_2d.py b/examples/annotator_2d.py
@@ -6,7 +6,8 @@
 from micro_sam.util import get_cache_directory
 
 DATA_CACHE = os.path.join(get_cache_directory(), "sample_data")
-EMBEDDING_CACHE = os.path.join(get_cache_directory, "embeddings")
+EMBEDDING_CACHE = os.path.join(get_cache_directory(), "embeddings")
+os.makedirs(EMBEDDING_CACHE, exist_ok=True)
 
 
 def livecell_annotator(use_finetuned_model):

diff --git a/examples/annotator_3d.py b/examples/annotator_3d.py
@@ -6,7 +6,8 @@
 from micro_sam.util import get_cache_directory
 
 DATA_CACHE = os.path.join(get_cache_directory(), "sample_data")
-EMBEDDING_CACHE = os.path.join(get_cache_directory, "embeddings")
+EMBEDDING_CACHE = os.path.join(get_cache_directory(), "embeddings")
+os.makedirs(EMBEDDING_CACHE, exist_ok=True)
 
 
 def em_3d_annotator(use_finetuned_model):

diff --git a/examples/annotator_tracking.py b/examples/annotator_tracking.py
@@ -6,7 +6,8 @@
 from micro_sam.util import get_cache_directory
 
 DATA_CACHE = os.path.join(get_cache_directory(), "sample_data")
-EMBEDDING_CACHE = os.path.join(get_cache_directory, "embeddings")
+EMBEDDING_CACHE = os.path.join(get_cache_directory(), "embeddings")
+os.makedirs(EMBEDDING_CACHE, exist_ok=True)
 
 
 def track_ctc_data(use_finetuned_model):

diff --git a/examples/image_series_annotator.py b/examples/image_series_annotator.py
@@ -5,7 +5,8 @@
 from micro_sam.util import get_cache_directory
 
 DATA_CACHE = os.path.join(get_cache_directory(), "sample_data")
-EMBEDDING_CACHE = os.path.join(get_cache_directory, "embeddings")
+EMBEDDING_CACHE = os.path.join(get_cache_directory(), "embeddings")
+os.makedirs(EMBEDDING_CACHE, exist_ok=True)
 
 
 def series_annotation(use_finetuned_model):