Skip to content

Commit 9ad1fe9

Browse files
author
Balaji Veeramani
committed
Merge branch 'serde-compatability' of github.com:bveeramani/sagemaker-python-sdk into serde-compatability
2 parents bbf6f2a + b9ce378 commit 9ad1fe9

File tree

1 file changed

+2
-5
lines changed
  • src/sagemaker/cli/compatibility/v2/modifiers

1 file changed

+2
-5
lines changed

src/sagemaker/cli/compatibility/v2/modifiers/serde.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def modify_node(self, node):
150150
return node
151151

152152
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]
154154
assert subpackage_name in {"serializers", "deserializers"}
155155
return pasta.parse("%s.%s()" % (subpackage_name, new_class_name)).body[0].value
156156

@@ -246,10 +246,7 @@ def modify_node(self, node):
246246
else:
247247
i += 1
248248

249-
if not node.names:
250-
return None
251-
252-
return node
249+
return node if node.names else None
253250

254251

255252
class SerdeImportFromAmazonCommonRenamer(Modifier):

0 commit comments

Comments
 (0)