Resolved build errors

evan-palmer · evan-palmer · commit 0d7087483ad6 · 2021-12-21T23:57:48.000-06:00
diff --git a/pymavswarm/__init__.py b/pymavswarm/__init__.py
@@ -1 +1,5 @@
-from .MavSwarm import MavSwarm
+from .MavSwarm import MavSwarm
+from .utils import MsgMap, OutgoingMsg
+from .Agent import Agent
+
+__all__ = ['MavSwarm', 'MsgMap', 'OutgoingMsg', 'Agent']
diff --git a/setup.py b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup
+import setuptools
 
 version = "0.0.1"
 
@@ -24,9 +25,7 @@
         'Topic :: Scientific/Engineering'
     ],
     license='GPLv3',
-    packages=[
-        'pymavswarm'
-    ],
+    packages=setuptools.find_packages(),
     setup_requires=[
         "setuptools", 
         "wheel"
