diff --git a/swarms/embeddings/pegasus.py b/swarms/embeddings/pegasus.py index de3a4e36..08cebba8 100644 --- a/swarms/embeddings/pegasus.py +++ b/swarms/embeddings/pegasus.py @@ -7,7 +7,12 @@ from pegasus import Pegasus class PegasusEmbedding: - def __init__(self, modality: str, multi_process: bool = False, n_processes: int = 4): + def __init__( + self, + modality: str, + multi_process: bool = False, + n_processes: int = 4 + ): self.modality = modality self.multi_process = multi_process self.n_processes = n_processes diff --git a/swarms/hivemind/hivemind.py b/swarms/hivemind/hivemind.py index a185c111..5ccf2432 100644 --- a/swarms/hivemind/hivemind.py +++ b/swarms/hivemind/hivemind.py @@ -13,7 +13,12 @@ from swarms.swarms.swarms import HierarchicalSwarm logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s') class HiveMind: - def __init__(self, openai_api_key="", num_swarms=1, max_workers=None): + def __init__( + self, + openai_api_key="", + num_swarms=1, + max_workers=None + ): self.openai_api_key = openai_api_key self.num_swarms = num_swarms self.swarms = [HierarchicalSwarm(openai_api_key) for _ in range(num_swarms)]