Removes dependency from MLJBase

ml-unito · Apr 8, 2024 · 8a28e0d · 8a28e0d
1 parent d177073
commit 8a28e0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/PartitionedLS.jl b/src/PartitionedLS.jl
@@ -226,7 +226,7 @@ It conforms to the MLJ interface.
 """
 
 function MLJModelInterface.predict(model::PartLS, fitresult, X)
-  X = MLJBase.matrix(X)
+  X = MLJModelInterface.matrix(X)
   return PartitionedLS.predict(fitresult, X)
 end