From 58103bbf14d453739327c5bf412aa0993d913ea3 Mon Sep 17 00:00:00 2001 From: chinganc Date: Tue, 19 Nov 2024 11:45:39 -0800 Subject: [PATCH] Update test_copy to omit testing when oai config is not found. --- tests/unit_tests/test_copy.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/unit_tests/test_copy.py b/tests/unit_tests/test_copy.py index e345349..d9b7dcc 100644 --- a/tests/unit_tests/test_copy.py +++ b/tests/unit_tests/test_copy.py @@ -18,10 +18,12 @@ def fun(x): x = trace.node('x', trainable=True) copy.deepcopy(x) -optimizer = OptoPrime([x]) -optimizer2 = copy.deepcopy(optimizer) + try: + optimizer = OptoPrime([x]) + optimizer2 = copy.deepcopy(optimizer) + llm = AutoGenLLM() copy.deepcopy(llm) except FileNotFoundError as e: