Add labels to metadata (#15)

panda-official · Jul 11, 2023 · 3514714 · 3514714
1 parent f652c9f
commit 3514714
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- Supports for labels in metadata, [PR-15](https://github.com/panda-official/DriftCLI/pull/15)
+
 ## 0.7.1 - 2023-06-29
 
 ### Fixed

diff --git a/drift_cli/export_impl/raw.py b/drift_cli/export_impl/raw.py
@@ -24,6 +24,7 @@ def _export_metadata_to_json(path: Path, pkg: DriftDataPackage):
             "status": pkg.status_code,
             "published_time": pkg.publish_timestamp,
             "source_timestamp": pkg.source_timestamp,
+            "labels": pkg.labels,
         }
         meta.update(MessageToDict(pkg.meta, preserving_proto_field_name=True))
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ requires = ["setuptools>=40.8.0", "wheel"]
 [project]
 
 name = "drift-cli"
-version = "0.7.1"
+version = "0.8.0"
 description = "CLI client for PANDA | Drift Platform"
 requires-python = ">=3.8"
 readme = "README.md"
@@ -29,7 +29,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "drift-python-client~=0.6.0",
+    "drift-python-client~=0.7.0",
     "click~=8.1",
     "tomlkit~=0.11",
     "rich~=12.6",

diff --git a/tests/export_test.py b/tests/export_test.py
@@ -182,6 +182,7 @@ def test__export_raw_data_with_metadata(
         metadata = json.load(file)
         assert metadata == {
             "id": 1,
+            "labels": {},
             "published_time": 0.0,
             "source_timestamp": 0.0,
             "status": 0,
@@ -306,6 +307,7 @@ def test__export_raw_data_topics_jpeg_with_metadata(
         metadata = json.load(file)
         assert metadata == {
             "id": 1,
+            "labels": {},
             "image_info": {"channel_layout": "RGB", "height": "100", "width": "100"},
             "published_time": 0.0,
             "source_timestamp": 0.0,