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: