Browse code

set learner._recompute_losses_factor = 1

Bas Nijholt authored on 13/04/2020 22:55:53
Showing 1 changed files
... ...
@@ -85,6 +85,8 @@ def test_serialization_for(learner_type, learner_kwargs, serializer):
85 85
         f = f_for_pickle  # noqa: F811
86 86
 
87 87
     learner = learner_type(f, **learner_kwargs)
88
+    if learner_type is Learner1D:
89
+        learner._recompute_losses_factor = 1
88 90
 
89 91
     simple(learner, goal_1)
90 92
     learner_bytes = serializer.dumps(learner)