Added integration tests

Author: vivianpengnyc

src/amp/admin/datasets.py

@@ -4,7 +4,7 @@
 including registration, deployment, versioning, and manifest operations.
 """
 
-from typing import TYPE_CHECKING, Dict, Optional
+from typing import TYPE_CHECKING, Dict, List, Optional, Union
 
 from amp.utils.manifest_inspector import describe_manifest, print_schema
 
@@ -200,7 +200,7 @@ def get_manifest(self, namespace: str, name: str, revision: str) -> dict:
         response = self._admin._request('GET', path)
         return response.json()
 
-    def describe(self, namespace: str, name: str, revision: str = 'latest') -> Dict[str, list[Dict[str, str | bool]]]:
+    def describe(self, namespace: str, name: str, revision: str = 'latest') -> Dict[str, List[Dict[str, Union[str, bool]]]]:
         """Get a structured summary of tables and columns in a dataset.
 
         Returns a dictionary mapping table names to lists of column information,

src/amp/registry/datasets.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from amp.utils.manifest_inspector import describe_manifest, print_schema
 
@@ -199,7 +199,7 @@ def get_manifest(self, namespace: str, name: str, version: str) -> dict:
         response = self._registry._request('GET', path)
         return response.json()
 
-    def describe(self, namespace: str, name: str, version: str = 'latest') -> Dict[str, list[Dict[str, str | bool]]]:
+    def describe(self, namespace: str, name: str, version: str = 'latest') -> Dict[str, List[Dict[str, Union[str, bool]]]]:
         """Get a structured summary of tables and columns in a dataset.
 
         Returns a dictionary mapping table names to lists of column information,

src/amp/streaming/state.py

@@ -9,7 +9,7 @@
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
 from datetime import UTC, datetime
-from typing import Dict, List, Optional, Set, Tuple
+from typing import Dict, List, Optional, Set, Tuple, Union
 
 from amp.streaming.types import BlockRange, ResumeWatermark
 
@@ -235,7 +235,7 @@ def __init__(self):
 
     def _get_key(
         self, connection_name: str, table_name: str, network: Optional[str] = None
-    ) -> Tuple[str, str, str] | List[Tuple[str, str, str]]:
+    ) -> Union[Tuple[str, str, str], List[Tuple[str, str, str]]]:
         """Get storage key(s) for the given parameters."""
         if network:
             return (connection_name, table_name, network)

src/amp/utils/manifest_inspector.py

@@ -4,10 +4,10 @@
 from manifest files in a human-readable format.
 """
 
-from typing import Any, Dict
+from typing import Any, Dict, List, Union
 
 
-def describe_manifest(manifest: dict) -> Dict[str, list[Dict[str, str | bool]]]:
+def describe_manifest(manifest: dict) -> Dict[str, List[Dict[str, Union[str, bool]]]]:
     """Extract structured schema information from a manifest.
 
     Args: