22
33
44def test_get_assets_returns_correct_artifact_for_logreg ():
5- module = LogregAimedEmbedding (k = 5 , embedder_name = "sergeyzh/rubert-tiny-turbo" )
5+ module = LogregAimedEmbedding (embedder_name = "sergeyzh/rubert-tiny-turbo" )
66 artifact = module .get_assets ()
77 assert artifact .embedder_name == "sergeyzh/rubert-tiny-turbo"
88
99
1010def test_fit_trains_model ():
11- module = LogregAimedEmbedding (k = 5 , embedder_name = "sergeyzh/rubert-tiny-turbo" )
11+ module = LogregAimedEmbedding (embedder_name = "sergeyzh/rubert-tiny-turbo" )
1212
1313 utterances = ["hello" , "goodbye" , "hi" , "bye" , "bye" , "hello" , "welcome" , "hi123" , "hiii" , "bye-bye" , "bye!" ]
1414 labels = [0 , 1 , 0 , 1 , 1 , 0 , 0 , 0 , 0 , 1 , 1 ]
@@ -20,7 +20,7 @@ def test_fit_trains_model():
2020
2121
2222def test_predict_evaluates_model ():
23- module = LogregAimedEmbedding (k = 5 , embedder_name = "sergeyzh/rubert-tiny-turbo" )
23+ module = LogregAimedEmbedding (embedder_name = "sergeyzh/rubert-tiny-turbo" )
2424
2525 utterances = ["hello" , "goodbye" , "hi" , "bye" , "bye" , "hello" , "welcome" , "hi123" , "hiii" , "bye-bye" , "bye!" ]
2626 labels = [0 , 1 , 0 , 1 , 1 , 0 , 0 , 0 , 0 , 1 , 1 ]
0 commit comments