File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
src/sagemaker/cli/compatibility/v2/modifiers Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -150,7 +150,7 @@ def modify_node(self, node):
150
150
return node
151
151
152
152
namespace_name = NEW_CLASS_NAME_TO_NAMESPACES [new_class_name ][0 ]
153
- subpackage_name = namespace_name [ namespace_name . find ("." ) + 1 : ]
153
+ subpackage_name = namespace_name . split ("." )[ 1 ]
154
154
assert subpackage_name in {"serializers" , "deserializers" }
155
155
return pasta .parse ("%s.%s()" % (subpackage_name , new_class_name )).body [0 ].value
156
156
@@ -246,10 +246,7 @@ def modify_node(self, node):
246
246
else :
247
247
i += 1
248
248
249
- if not node .names :
250
- return None
251
-
252
- return node
249
+ return node if node .names else None
253
250
254
251
255
252
class SerdeImportFromAmazonCommonRenamer (Modifier ):
You can’t perform that action at this time.
0 commit comments