From 9c87e6d4da7a4dd08220cd0ec436a3651bc571af Mon Sep 17 00:00:00 2001
From: Serghei Cebotari <serghei@cebotari.ru>
Date: Thu, 11 Jan 2024 17:48:01 +0300
Subject: Add ModelBuilder

---
 RhSolutions.ML.Tests/TestBase.cs | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

(limited to 'RhSolutions.ML.Tests/TestBase.cs')

diff --git a/RhSolutions.ML.Tests/TestBase.cs b/RhSolutions.ML.Tests/TestBase.cs
index 679d9d3..6295c3f 100644
--- a/RhSolutions.ML.Tests/TestBase.cs
+++ b/RhSolutions.ML.Tests/TestBase.cs
@@ -1,5 +1,3 @@
-using RhSolutions.ML.Lib;
-
 namespace RhSolutions.ML.Tests;
 
 public abstract class TestBase
@@ -11,7 +9,6 @@ public abstract class TestBase
 	
 	public TestBase()
 	{
-		RhSolutionsMLBuilder.RebuildModel();
 		_mlContext = new MLContext(seed: 0);
 		ITransformer loadedNodel = _mlContext.Model.Load(_dataPath, out var _);
 		_predEngine = _mlContext.Model.CreatePredictionEngine<Product, TypePrediction>(loadedNodel);
@@ -36,4 +33,4 @@ public abstract class TestBase
 		var prediction = _predEngine.Predict(actual);
 		Assert.That(prediction.Type, Is.EqualTo(expected.Type));
 	}
-}
\ No newline at end of file
+}
-- 
cgit v1.2.3