Slight relative path import fix

guydav · Nov 28, 2022 · 2e28982 · 2e28982
1 parent 47b70ae
commit 2e28982
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/simple_relational_reasoning/embeddings/models.py b/simple_relational_reasoning/embeddings/models.py
@@ -8,7 +8,9 @@
 import torchvision.models as models
 
 sys.path.append(os.path.abspath('.'))
+sys.path.append(os.path.abspath('./silicon_menagerie'))
 sys.path.append(os.path.abspath('..'))
+sys.path.append(os.path.abspath('../silicon_menagerie'))
 from silicon_menagerie.utils import load_model as emin_load_model
 
 CHECKPOINT_FOLDER = r'/home/gd1279/scratch/SAYcam-models'