From a385c0bb631d3ae36291802771d79db2589eeaea Mon Sep 17 00:00:00 2001
From: AnnaTz <111577222+AnnaTz@users.noreply.github.com>
Date: Thu, 30 Nov 2023 17:26:49 +0000
Subject: [PATCH] refactor: Avoid changing the backend in KerasModule's
 __call__ if it's already tensorflow.

---
 ivy/stateful/converters.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/ivy/stateful/converters.py b/ivy/stateful/converters.py
index 13b5e5cfde568..599b5532d0a24 100644
--- 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):