Added setter to change one a session's data paths

model_server/session.py

@@ -36,6 +36,13 @@ class Session(object):
     def get_paths(self):
         return self.paths
 
+    def set_data_directory(self, key: str, path: Path):
+        if not key in self.paths.keys():
+            raise InvalidPathError(f'No such path {key}')
+        if not Path(path).exists():
+            raise InvalidPathError(f'Could not find {path}')
+        self.paths[key] = path
+
     @staticmethod
     def make_paths(root: str = None) -> dict:
         """
@@ -143,4 +150,7 @@ class CouldNotInstantiateModelError(Error):
     pass
 
 class CouldNotCreateDirectory(Error):
+    pass
+
+class InvalidPathError(Error):
     pass
\ No newline at end of file

tests/test_session.py

@@ -28,6 +28,13 @@ class TestGetSessionObject(unittest.TestCase):
         rmtree(newroot)
         self.assertFalse(newroot.exists(), 'Could not clean up temporary test subdirectory')
 
+    def test_change_session_subdirectory(self):
+        sesh = Session()
+        old_paths = sesh.get_paths()
+        print(old_paths)
+        sesh.set_data_directory('outbound_images', old_paths['inbound_images'])
+        self.assertEqual(sesh.paths['outbound_images'], sesh.paths['inbound_images'])
+
 
     def test_restart_session(self):
         sesh = Session()