add missing module imports

psyplot · Sep 29, 2019 · bb33c46 · bb33c46
1 parent d3a3e22
commit bb33c46
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/psy_reg/utils.py b/psy_reg/utils.py
@@ -1,5 +1,10 @@
+"""Utility functions for psy-reg"""
 import abc
+import inspect
+from itertools import cycle
 from scipy.optimize import curve_fit, differential_evolution
+import numpy as np
+from warnings import warn
 
 
 def rsquared(sim, obs):
@@ -115,4 +120,4 @@ def fit(cls, x, y, *args, **kwargs):
         if pcov.size == 1:
             attrs['err'] = np.sqrt(pcov)[0, 0]
 
-        return cls(*params, **attrs)
+        return cls(*params, **attrs)