Skip to content

Commit 30ea8c4

Browse files
authored
Remove unnecessary metadata fields in UUID materializer test (#3088)
1 parent f042cc0 commit 30ea8c4

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

src/zenml/materializers/uuid_materializer.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,5 @@ def extract_metadata(self, data: uuid.UUID) -> Dict[str, MetadataType]:
7575
A dictionary of metadata extracted from the UUID.
7676
"""
7777
return {
78-
"uuid_version": str(data.version),
79-
"uuid_variant": data.variant,
8078
"string_representation": str(data),
8179
}

tests/unit/materializers/test_uuid_materializer.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) ZenML GmbH 2023. All Rights Reserved.
1+
# Copyright (c) ZenML GmbH 2024. All Rights Reserved.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.
@@ -26,7 +26,7 @@ def test_uuid_materializer():
2626
step_output_type=uuid.UUID,
2727
materializer_class=UUIDMaterializer,
2828
step_output=test_uuid,
29-
expected_metadata_size=4,
29+
expected_metadata_size=2,
3030
)
3131
assert result == test_uuid
3232
assert isinstance(result, uuid.UUID)
@@ -37,7 +37,5 @@ def test_uuid_materializer_metadata():
3737
test_uuid = uuid.uuid4()
3838
materializer = UUIDMaterializer("test_uri")
3939
metadata = materializer.extract_metadata(test_uuid)
40-
assert len(metadata) == 3
41-
assert metadata["uuid_version"] == test_uuid.version
42-
assert metadata["uuid_variant"] == test_uuid.variant
40+
assert len(metadata) == 1
4341
assert metadata["string_representation"] == str(test_uuid)

0 commit comments

Comments
 (0)