Merge pull request #319 from facebookresearch/change-other-world-loading

Change how the ParlAI chat world is loaded throughout
facebookresearch · Dec 1, 2020 · 1fcc45d · 1fcc45d
2 parents b23d5b5 + 581dea5
commit 1fcc45d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_blueprint.py b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_blueprint.py
@@ -165,8 +165,8 @@ def __init__(
 
         if shared_state.world_module is None:
             world_file_path = os.path.expanduser(args.blueprint.world_file)
-            world_module_path = world_file_path[:-3]
-            sys.path.append(world_module_path)
+            world_module_dir = os.path.dirname(world_file_path)
+            sys.path.append(world_module_dir)
             world_module_name = os.path.basename(world_file_path)[:-3]
             world_module = import_module(world_module_name)
         else:

diff --git a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_task_runner.py b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_task_runner.py
@@ -129,8 +129,8 @@ def __init__(
     ):
         super().__init__(task_run, args, shared_state)
         world_file_path = os.path.expanduser(args.blueprint.world_file)
-        world_module_path = world_file_path[:-3]
-        sys.path.append(world_module_path)
+        world_module_dir = os.path.dirname(world_file_path)
+        sys.path.append(world_module_dir)
         world_module_name = os.path.basename(world_file_path)[:-3]
         self.parlai_world_module = import_module(world_module_name)
         world_params = self.parlai_world_module.get_world_params()