From 2e28982d0ba53d8d211819229df31e259426a2ce Mon Sep 17 00:00:00 2001
From: Guy Davidson <guydav@gmail.com>
Date: Mon, 28 Nov 2022 12:26:41 +0200
Subject: [PATCH] Slight relative path import fix

---
 simple_relational_reasoning/embeddings/models.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/simple_relational_reasoning/embeddings/models.py b/simple_relational_reasoning/embeddings/models.py
index 533c04c..19b7a21 100644
--- 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'