fix import

edahelsinki · Feb 16, 2024 · 63bb311 · 63bb311
1 parent 4fe51c7
commit 63bb311
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_local_models.py b/tests/test_local_models.py
@@ -1,12 +1,12 @@
-from statistics import LinearRegression, linear_regression
-
 import numpy as np
 import pytest
 import torch
 
 from slisemap.local_models import (
     ALocalModel,
+    LinearRegression,
     identify_local_model,
+    linear_regression,
     linear_regression_coefficients,
     linear_regression_loss,
     local_predict,