From 1f7b2f2f72ab80e0cb8c67fa1ef03844391bd022 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 19:47:57 -0400 Subject: [PATCH] clean up --- swarms/orchestrate.py | 4 ++-- tests/agents/workers/worker_ultra.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/swarms/orchestrate.py b/swarms/orchestrate.py index f869fc3c..63ac5c10 100644 --- a/swarms/orchestrate.py +++ b/swarms/orchestrate.py @@ -44,10 +44,10 @@ Orchestrate(WorkerNode, autoscale=True, nodes=int, swarm_type="flat") """ #in a shared state, provide 3 communication times, during task assignment, task compeltion, and feedback or inability to complete a task. +import threading from abc import ABC, abstractmethod -from typing import List, Dict, Any +from typing import Any, Dict, List -import threading from swarms.agents.memory.ocean import OceanDB diff --git a/tests/agents/workers/worker_ultra.py b/tests/agents/workers/worker_ultra.py index 94aca5e3..17b699e3 100644 --- a/tests/agents/workers/worker_ultra.py +++ b/tests/agents/workers/worker_ultra.py @@ -1,6 +1,6 @@ import pytest from unittest.mock import Mock, patch -from swarms.workers.worker_agent_ultra import WorkerUltraNode, WorkerUltraNodeInitializer # replace 'yourmodule' with the actual module name +from swarms.workers.worker_agent_ultra import WorkerUltraNode, WorkerUltraNodeInitializer @pytest.fixture def llm_mock():