fix: minor fix to suffix variable names with :0

ivy-llc · Jul 26, 2024 · fa4ed6b · fa4ed6b
1 parent 0a357b6
commit fa4ed6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/stateful/module.py b/ivy/stateful/module.py
@@ -778,7 +778,7 @@ def _build(self, *args, **kwargs):
         import tensorflow as tf
 
         def _get_variable_name(variable):
-            return variable.path.split("/")[-2] + "/" + variable.name
+            return variable.path.split("/")[-2] + "/" + variable.name + ":0"
 
         self._native_params = ivy.Container(
             OrderedDict(