fixed config file parsing bug

facebookresearch · Jun 22, 2019 · a27649d · a27649d
1 parent 303d87c
commit a27649d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pyrobot/core.py b/src/pyrobot/core.py
@@ -72,8 +72,8 @@ def __init__(self,
         cfg_path = os.path.join(root_path, 'cfg')
         robot_pool = []
         for f in os.listdir(cfg_path):
-            if '_config.py' in f:
-                robot_pool.append(f.strip('_config.py'))
+            if f.endswith('_config.py'):
+                robot_pool.append(f[:-len('_config.py')])
         root_node = 'pyrobot.'
         self.configs = None
         this_robot = None