fix: address review comments (iteration #1)

Author: aviruthen

sagemaker-core/src/sagemaker/core/utils/utils.py

@@ -273,7 +273,7 @@ def pascal_to_snake(pascal_str):
 
 
 def is_not_primitive(obj):
-        return not isinstance(obj, (int, float, str, bool, bytes, datetime.datetime))
+    return not isinstance(obj, (int, float, str, bool, bytes, datetime.datetime))
 
 
 def is_not_str_dict(obj):
@@ -285,7 +285,7 @@ def is_primitive_list(obj):
 
 
 def is_primitive_class(cls):
-        return cls in (str, int, bool, float, bytes, datetime.datetime)
+    return cls in (str, int, bool, float, bytes, datetime.datetime)
 
 
 class Unassigned:

sagemaker-core/tests/unit/generated/test_utils.py

@@ -374,29 +374,20 @@ def test_serialize_method_nested_shape():
 
 
 
-def test_serialize_with_bytes_value_returns_bytes():
-    result = serialize(b'1')
-    assert result == b'1'
-
-
-def test_serialize_dict_with_bytes_value_returns_dict_with_bytes():
+def test_serialize_bytes_returns_bytes():
     result = serialize({'body': b'1'})
     assert result == {'body': b'1'}
+    assert serialize(b'hello') == b'hello'
 
 
-def test_is_not_primitive_with_bytes_returns_false():
-    assert is_not_primitive(b'hello') is False
+def test_is_not_primitive_with_bytes():
+    assert is_not_primitive(b'test') is False
 
 
-def test_is_primitive_class_with_bytes_returns_true():
+def test_is_primitive_class_with_bytes():
     assert is_primitive_class(bytes) is True
 
 
-def test_serialize_list_with_bytes_elements():
-    result = serialize([b'a', b'b'])
-    assert result == [b'a', b'b']
-
-
 class TestUnassignedBehavior:
     """Test Unassigned class methods for proper behavior.