From a92aafaf9423d3bebf72392528879f1e5d4fac67 Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Wed, 22 Nov 2023 11:50:47 -0700 Subject: [PATCH 1/2] ruff cleanup - unused imports --- playground/models/kosmos2.py | 2 +- swarms/models/simple_ada.py | 1 - swarms/structs/flow.py | 1 - swarms/utils/apa.py | 2 -- swarms/utils/main.py | 1 - swarms/utils/pdf_to_text.py | 1 - tests/models/bioclip.py | 1 - 7 files changed, 1 insertion(+), 8 deletions(-) diff --git a/playground/models/kosmos2.py b/playground/models/kosmos2.py index ce39a710..6fc4df02 100644 --- a/playground/models/kosmos2.py +++ b/playground/models/kosmos2.py @@ -1,4 +1,4 @@ -from swarms.models.kosmos2 import Kosmos2, Detections +from swarms.models.kosmos2 import Kosmos2 from PIL import Image diff --git a/swarms/models/simple_ada.py b/swarms/models/simple_ada.py index 3662dda2..6a0dbcc9 100644 --- a/swarms/models/simple_ada.py +++ b/swarms/models/simple_ada.py @@ -1,4 +1,3 @@ -import os from openai import OpenAI client = OpenAI() diff --git a/swarms/structs/flow.py b/swarms/structs/flow.py index fd359592..81f05cdf 100644 --- a/swarms/structs/flow.py +++ b/swarms/structs/flow.py @@ -3,7 +3,6 @@ import inspect import json import logging import random -import re import time from typing import Any, Callable, Dict, List, Optional, Tuple diff --git a/swarms/utils/apa.py b/swarms/utils/apa.py index 94c6f158..9705d93c 100644 --- a/swarms/utils/apa.py +++ b/swarms/utils/apa.py @@ -1,7 +1,5 @@ from enum import Enum, unique, auto import abc -import hashlib -import re from typing import List, Optional import json from dataclasses import dataclass, field diff --git a/swarms/utils/main.py b/swarms/utils/main.py index a6c4fc34..cd21b456 100644 --- a/swarms/utils/main.py +++ b/swarms/utils/main.py @@ -9,7 +9,6 @@ from typing import Dict import boto3 import numpy as np -import pandas as pd import requests diff --git a/swarms/utils/pdf_to_text.py b/swarms/utils/pdf_to_text.py index b8778841..f3e7105a 100644 --- a/swarms/utils/pdf_to_text.py +++ b/swarms/utils/pdf_to_text.py @@ -1,5 +1,4 @@ import sys -import os try: import PyPDF2 diff --git a/tests/models/bioclip.py b/tests/models/bioclip.py index 50a65570..383d96ac 100644 --- a/tests/models/bioclip.py +++ b/tests/models/bioclip.py @@ -1,5 +1,4 @@ # Import necessary modules and define fixtures if needed -import os import pytest import torch from PIL import Image From 06e2294e5eb244877ab62a9968a2145f16a28b6e Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Wed, 22 Nov 2023 12:23:11 -0700 Subject: [PATCH 2/2] ruff lint - remove duplicate code --- tests/swarms/multi_agent_collab.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/tests/swarms/multi_agent_collab.py b/tests/swarms/multi_agent_collab.py index 3f7a0729..3a8d72cb 100644 --- a/tests/swarms/multi_agent_collab.py +++ b/tests/swarms/multi_agent_collab.py @@ -93,21 +93,12 @@ def test_performance(collaboration): assert agent.name in performance_data assert "metrics" in performance_data[agent.name] - def test_set_interaction_rules(collaboration): rules = {"rule1": "action1", "rule2": "action2"} collaboration.set_interaction_rules(rules) assert hasattr(collaboration, "interaction_rules") assert collaboration.interaction_rules == rules - -def test_set_interaction_rules(collaboration): - rules = {"rule1": "action1", "rule2": "action2"} - collaboration.set_interaction_rules(rules) - assert hasattr(collaboration, "interaction_rules") - assert collaboration.interaction_rules == rules - - def test_repr(collaboration): repr_str = repr(collaboration) assert isinstance(repr_str, str)