diff --git a/course/page/code_feedback.py b/course/page/code_feedback.py index 257a8b575..c2508bc54 100644 --- a/course/page/code_feedback.py +++ b/course/page/code_feedback.py @@ -64,7 +64,9 @@ def check_numpy_array_sanity(self, name, num_axes, data): 0, f"'{name}' does not consist of floating point numbers--" f"got: '{data.dtype}'") - def check_numpy_array_features(self, name, ref, data, report_failure=True, check_finite=True): + def check_numpy_array_features(self, name, ref, data, check_finite=True, + report_failure=True): + import numpy as np assert isinstance(ref, np.ndarray)