fix test

yoelcortes · May 24, 2024 · ea2386a · ea2386a
1 parent dcc6b0b
commit ea2386a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flexsolve/profiler.py b/flexsolve/profiler.py
@@ -72,7 +72,10 @@ def __call__(self, x, *args):
         x = copy(x)
         self.xs.append(x)
         y = self.f(x, *args)
-        self.ys.append(y)
+        if isinstance(y, tuple): 
+            self.ys.append(y[0])
+        else:
+            self.ys.append(y)
         return y
 
     def archive(self, name):