update import statements for external usage

Author: zehuilu

lib/CBBA.py

@@ -4,8 +4,8 @@
 import copy
 import matplotlib.pyplot as plt
 import numpy as np
-from lib.Task import Task
-from lib.WorldInfo import WorldInfo
+from Task import Task
+from WorldInfo import WorldInfo
 
 
 class CBBA(object):

lib/HelperLibrary.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 import random
 import numpy as np
-from lib.Agent import Agent
-from lib.Task import Task
-from lib.WorldInfo import WorldInfo
+from Agent import Agent
+from Task import Task
+from WorldInfo import WorldInfo
 
 
 def create_agents_and_tasks(num_agents: int, num_tasks: int, WorldInfoInput: WorldInfo, config_data):

test/pathmagic.py

@@ -7,7 +7,7 @@ class context:
     def __enter__(self):
         # append the main directory of CBBA-Python to sys.path
         # assume the working directory is always the main directory of CBBA-Python
-        sys.path.append(os.getcwd())
+        sys.path.append(os.getcwd() + "/lib")
         print(os.getcwd())
 
     def __exit__(self, *args):

test/run_cbba_example_01.py

@@ -4,9 +4,9 @@
 import matplotlib.pyplot as plt
 import pathmagic
 with pathmagic.context():
-    from lib.CBBA import CBBA
-    from lib.WorldInfo import WorldInfo
-    import lib.HelperLibrary as HelperLibrary
+    from CBBA import CBBA
+    from WorldInfo import WorldInfo
+    import HelperLibrary as HelperLibrary
 
 
 if __name__ == "__main__":

test/run_cbba_example_02.py

@@ -4,9 +4,9 @@
 import matplotlib.pyplot as plt
 import pathmagic
 with pathmagic.context():
-    from lib.CBBA import CBBA
-    from lib.WorldInfo import WorldInfo
-    import lib.HelperLibrary as HelperLibrary
+    from CBBA import CBBA
+    from WorldInfo import WorldInfo
+    import HelperLibrary as HelperLibrary
 
 
 if __name__ == "__main__":

test/run_cbba_example_03.py

@@ -4,9 +4,9 @@
 import matplotlib.pyplot as plt
 import pathmagic
 with pathmagic.context():
-    from lib.CBBA import CBBA
-    from lib.WorldInfo import WorldInfo
-    import lib.HelperLibrary as HelperLibrary
+    from CBBA import CBBA
+    from WorldInfo import WorldInfo
+    import HelperLibrary as HelperLibrary
 
 
 if __name__ == "__main__":