Revert modelpath for environment to be related to datapath.

berkeley-reclab · Jun 29, 2022 · 57ac291 · 57ac291
1 parent 8546859
commit 57ac291
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/reclab/data_utils.py b/reclab/data_utils.py
@@ -10,7 +10,7 @@
 
 DATA_DIR = os.environ.get('RECLAB_DATA_PATH')
 if DATA_DIR is None:
-    DATA_DIR = os.path.expanduser('~/recsys/data/')
+    DATA_DIR = os.path.dirname(__file__)
 
 def read_dataset(name, shuffle=True, seed=0):
     """Read a dataset as specified by name.

diff --git a/reclab/environments/latent_factors.py b/reclab/environments/latent_factors.py
@@ -214,7 +214,7 @@ def __init__(self, name, latent_dim=128, datapath=data_utils.DATA_DIR, force_ret
                  max_num_users=np.inf, max_num_items=np.inf, **kwargs):
         """Create a ML100K Latent Factor environment."""
         self.dataset_name = name
-        modelpath = os.path.join(os.path.dirname(__file__), '../../models')
+        modelpath = datapath
         if name == 'ml-100k':
             self.datapath = os.path.expanduser(os.path.join(datapath, 'ml-100k'))
             self.modelpath = os.path.join(modelpath, 'ml-100k')