From 8305c14e605ec7aa1e9379609f80f29e41453db4 Mon Sep 17 00:00:00 2001 From: Kye Date: Sun, 3 Sep 2023 22:37:22 -0400 Subject: [PATCH] clean up Former-commit-id: 99f60ac9bec14fdc3dd77a8b4a09f4d59d303400 --- swarms/embeddings/pegasus.py | 7 ++++++- swarms/hivemind/hivemind.py | 7 ++++++- 2 files changed, 12 insertions(+), 2 deletions(-) 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)]