pkg issue fixed

.gitignore

@@ -5,4 +5,5 @@ __pycache__/
 venv/
 .DS_Store
 **.egg-info/
-.VSCodeCounter/
\ No newline at end of file
+.VSCodeCounter/
+dist/dryvr_plus_plus-0.1-py3.8.egg

demo/demo1.py

@@ -3,7 +3,7 @@ from dryvr_plus_plus.scene_verifier.scenario.scenario import Scenario
 from dryvr_plus_plus.example.example_map.simple_map2 import SimpleMap2, SimpleMap3, SimpleMap5, SimpleMap6
 from dryvr_plus_plus.plotter.plotter2D import *
 from dryvr_plus_plus.example.example_sensor.fake_sensor import FakeSensor2
-
+import plotly.graph_objects as go
 import matplotlib.pyplot as plt
 import numpy as np
 from enum import Enum, auto
@@ -55,7 +55,7 @@ if __name__ == "__main__":
             (VehicleMode.Normal, LaneMode.Lane1),
         ]
     )
-    # res_list = scenario.simulate_multi(10,1)
+    # res_list = scenario.simulate_multi(10, 1)
     # # traces = scenario.verify(10)
 
     # fig = plt.figure(2)
@@ -63,16 +63,18 @@ if __name__ == "__main__":
     # # fig = plot_reachtube_tree(traces, 'car1', 0, [1], 'b', fig, (1000,-1000), (1000,-1000))
     # # fig = plot_reachtube_tree(traces, 'car2', 0, [1], 'r', fig)
     # for traces in res_list:
-    #     # fig = plot_simulation_tree(traces, 'car1', 0, [1], 'b', fig, (1000,-1000), (1000,-1000))
-    #     # fig = plot_simulation_tree(traces, 'car2', 0, [1], 'r', fig)
-    #     generate_simulation_anime(traces, tmp_map, fig)
+    #     fig = plot_simulation_tree(
+    #         traces, 'car1', 1, [2], 'b', fig, (1000, -1000), (1000, -1000))
+    #     fig = plot_simulation_tree(traces, 'car2', 1, [2], 'r', fig)
+    #     # generate_simulation_anime(traces, tmp_map, fig)
 
     # plt.show()
 
     traces = scenario.simulate(10)
     fig = go.Figure()
-    fig = plotly_map(tmp_map, 'g', fig)
-    fig = plot_simulation_tree(
-        traces, 'car1', 0, [1], 'b', fig, (1000, -1000), (1000, -1000))
-    fig = plot_simulation_tree(traces, 'car2', 0, [1], 'r', fig)
+    # fig = plotly_map(tmp_map, 'g', fig)
+    # fig = plotly_simulation_tree(
+    #     traces, 'car1', 0, [1], 'b', fig, (1000, -1000), (1000, -1000))
+    # fig = plotly_simulation_tree(traces, 'car2', 0, [1], 'r', fig)
+    fig = plotly_simulation_anime(traces, tmp_map, fig)
     fig.show()

demo/demo2.py

@@ -3,7 +3,7 @@ from dryvr_plus_plus.scene_verifier.scenario.scenario import Scenario
 from dryvr_plus_plus.example.example_map.simple_map2 import SimpleMap2, SimpleMap3, SimpleMap5, SimpleMap6
 from dryvr_plus_plus.plotter.plotter2D import *
 from dryvr_plus_plus.example.example_sensor.fake_sensor import FakeSensor2
-
+import plotly.graph_objects as go
 import matplotlib.pyplot as plt
 import numpy as np
 from enum import Enum, auto

setup.py

 #!/usr/bin/env python3
-from setuptools import setup
+from setuptools import setup, find_packages
 
 setup(
     name='dryvr_plus_plus',
@@ -9,7 +9,7 @@ setup(
     maintainer='Yangge Li, Katherine Braught, Haoqing Zhu',
     maintainer_email='{li213, braught2, haoqing3}@illinois.edu',
     license='Apache-2.0',
-    packages=setuptools.find_packages(exclude=["tests", "demo"]),
+    packages=find_packages(exclude=["tests", "demo"]),
     python_requires='>=3.8',
     install_requires=[
         "numpy~=1.22.1",
@@ -24,6 +24,7 @@ setup(
         "treelib~=1.6.1",
         "z3-solver~=4.8.17.0",
         "igraph~=0.9.10",
+        "plotly~=5.8.0"
     ],
     classifiers=[
         'Development Status :: 2 - Pre-Alpha',
@@ -32,4 +33,4 @@ setup(
         'License :: OSI Approved :: Apache License 2.0',
         'Programming Language :: Python :: 3.8',
     ]
-)
\ No newline at end of file
+)