chore: fix styling

jina-ai · Oct 6, 2021 · d970a2b · d970a2b
1 parent e62f77e
commit d970a2b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/integration/torch/test_torch_trainer.py b/tests/integration/torch/test_torch_trainer.py
@@ -6,7 +6,7 @@
 import torch.nn as nn
 
 from finetuner.tuner.pytorch import PytorchTuner
-from finetuner.toydata import generate_fashion_match as fmdg
+from finetuner.toydata import generate_fashion_match
 from finetuner.toydata import generate_qa_match
 
 
@@ -27,8 +27,10 @@ def test_simple_sequential_model(tmpdir, params, head_layer):
 
     # fit and save the checkpoint
     pt.fit(
-        train_data=lambda: fmdg(num_pos=10, num_neg=10, num_total=params['num_train']),
-        eval_data=lambda: fmdg(
+        train_data=lambda: generate_fashion_match(
+            num_pos=10, num_neg=10, num_total=params['num_train']
+        ),
+        eval_data=lambda: generate_fashion_match(
             num_pos=10, num_neg=10, num_total=params['num_eval'], is_testset=True
         ),
         epochs=params['epochs'],