refactor: Avoid changing the backend in KerasModule's __call__ if it'…

…s already tensorflow.
mobley-trent · Nov 30, 2023 · a385c0b · a385c0b
1 parent 423ebed
commit a385c0b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/ivy/stateful/converters.py b/ivy/stateful/converters.py
@@ -527,10 +527,12 @@ def call(self, *args, training=None, **kwargs):
                 return ret
 
             def __call__(self, *args, **kwargs):
-                ivy.set_backend("tensorflow")
-                args, kwargs = ivy.args_to_new_backend(*args, native=True, **kwargs)
-                ivy.previous_backend()
-
+                if ivy.backend != "tensorflow":
+                    ivy.set_backend("tensorflow")
+                    args, kwargs = ivy.args_to_new_backend(*args, native=True, **kwargs)
+                    ivy.previous_backend()
+                else:
+                    args, kwargs = ivy.args_to_new_backend(*args, native=True, **kwargs)
                 return super(KerasModel, self).__call__(*args, **kwargs)
 
             def to_device(self, device):