revert test

Author: divyashreepathihalli

integration_tests/basic_full_flow.py

@@ -52,8 +52,8 @@ 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(x)
         model.predict(x)
+        model(x)
 
     @pytest.mark.skipif(
         os.environ.get("KERAS_NNX_ENABLED") != "true",