clean up

setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 setup(
   name = 'swarms',
   packages = find_packages(exclude=[]),
-  version = '0.5.6',
+  version = '0.5.7',
   license='MIT',
   description = 'Swarms - Pytorch',
   author = 'Kye Gomez',

swarms/__init__.py

@@ -1 +1 @@
-from swarms.agents.swarms import Swarms
+from swarms.swarms import Swarms

swarms/agents/__init__.py

@@ -0,0 +1 @@
+from ..swarms import Swarms

unittests/swarms.py

@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import patch
 # from swarms.agents.swarms import llm, tools, vectorstore, task_execution_chain, WorkerNode, BossNode
-from swarms.agents.swarms import worker_node, boss_node
+from swarms.swarms import worker_node, boss_node
 
 
 class TestSwarms(unittest.TestCase):