fix test

Author: divyashreepathihalli

integration_tests/basic_full_flow.py

@@ -50,5 +50,5 @@ def test_basic_fit(self):
     def test_basic_fit_no_training(self):
         model = MyModel(hidden_dim=2, output_dim=1)
         x = np.random.random((128, 4))
-        model.predict(x)
         model(x)
+        model.predict(x)