Polish code

PaddlePaddle · Jul 17, 2023 · 9d63ce5 · 9d63ce5
1 parent 94b568f
commit 9d63ce5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/sot/opcode_translator/executor/variables/callable.py b/sot/opcode_translator/executor/variables/callable.py
@@ -299,9 +299,7 @@ def call_function(self, /, *args, **kwargs):
 
     @VariableFactory.register_from_value(successor="PaddleApiVariable")
     def from_value(value: Any, graph: FunctionGraph, tracker: Tracker):
-        if isinstance(
-            value, paddle.nn.Layer
-        ) and not value.__module__.startswith("paddle.nn."):
+        if isinstance(value, paddle.nn.Layer):
             return UserDefinedLayerVariable(value, graph, tracker)
         return None