pull/170/merge
evelynmitchell 1 year ago committed by GitHub
commit 38da174b1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,4 +1,4 @@
from swarms.models.kosmos2 import Kosmos2, Detections from swarms.models.kosmos2 import Kosmos2
from PIL import Image from PIL import Image

@ -1,4 +1,3 @@
import os
from openai import OpenAI from openai import OpenAI
client = OpenAI() client = OpenAI()

@ -3,7 +3,6 @@ import inspect
import json import json
import logging import logging
import random import random
import re
import time import time
from typing import Any, Callable, Dict, List, Optional, Tuple from typing import Any, Callable, Dict, List, Optional, Tuple

@ -1,7 +1,5 @@
from enum import Enum, unique, auto from enum import Enum, unique, auto
import abc import abc
import hashlib
import re
from typing import List, Optional from typing import List, Optional
import json import json
from dataclasses import dataclass, field from dataclasses import dataclass, field

@ -9,7 +9,6 @@ from typing import Dict
import boto3 import boto3
import numpy as np import numpy as np
import pandas as pd
import requests import requests

@ -1,5 +1,4 @@
import sys import sys
import os
try: try:
import PyPDF2 import PyPDF2

@ -1,5 +1,4 @@
# Import necessary modules and define fixtures if needed # Import necessary modules and define fixtures if needed
import os
import pytest import pytest
import torch import torch
from PIL import Image from PIL import Image

@ -93,21 +93,12 @@ def test_performance(collaboration):
assert agent.name in performance_data assert agent.name in performance_data
assert "metrics" in performance_data[agent.name] assert "metrics" in performance_data[agent.name]
def test_set_interaction_rules(collaboration): def test_set_interaction_rules(collaboration):
rules = {"rule1": "action1", "rule2": "action2"} rules = {"rule1": "action1", "rule2": "action2"}
collaboration.set_interaction_rules(rules) collaboration.set_interaction_rules(rules)
assert hasattr(collaboration, "interaction_rules") assert hasattr(collaboration, "interaction_rules")
assert collaboration.interaction_rules == 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): def test_repr(collaboration):
repr_str = repr(collaboration) repr_str = repr(collaboration)
assert isinstance(repr_str, str) assert isinstance(repr_str, str)

Loading…
Cancel
Save