Normalize imports to match other modules

triton-inference-server · Feb 6, 2025 · e2b99ae · e2b99ae
1 parent 70063a0
commit e2b99ae
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 6 deletions.
diff --git a/icp/python/src/triton_distributed/icp/event_plane.py b/icp/python/src/triton_distributed/icp/event_plane.py
@@ -55,7 +55,7 @@ def __init__(self, event_topic: Union[List[str], str]):
                 "Invalid event_topic string. Only alphanumeric characters, underscores, and hyphens are allowed."
             )
         event_topic = ".".join(event_topic)
-        super().__init__(event_topic=event_topic)
+        self.event_topic = event_topic
 
     def __str__(self):
         return self.event_topic

diff --git a/icp/python/src/triton_distributed/icp/nats_event_plane.py b/icp/python/src/triton_distributed/icp/nats_event_plane.py
@@ -19,7 +19,7 @@
 
 import nats
 
-from triton_distributed.icp import EventMetadata, EventTopic
+from triton_distributed.icp.event_plane import EventMetadata, EventTopic
 
 
 class NatsEventPlane:

diff --git a/icp/tests/python/event_plane/test_basics.py b/icp/tests/python/event_plane/test_basics.py
@@ -19,12 +19,16 @@
 
 import pytest
 
-from triton_distributed.icp import EventMetadata, EventTopic, NatsEventPlane
+from triton_distributed.icp.nats_event_plane import (
+    EventMetadata,
+    EventTopic,
+    NatsEventPlane,
+)
 
 
 class TestEventTopic:
     def test_from_string(self):
-        topic_str = "level1.level2"
+        topic_str = "level1"
         event_topic = EventTopic(topic_str)
         assert event_topic.event_topic == topic_str
 

diff --git a/icp/tests/python/event_plane/test_publish_subscribe.py b/icp/tests/python/event_plane/test_publish_subscribe.py
@@ -20,7 +20,11 @@
 
 import pytest
 
-from triton_distributed.icp import EventMetadata, EventTopic, NatsEventPlane
+from triton_distributed.icp.nats_event_plane import (
+    EventMetadata,
+    EventTopic,
+    NatsEventPlane,
+)
 
 
 @pytest.mark.asyncio

diff --git a/icp/tests/python/event_plane/test_specific_subscriptions.py b/icp/tests/python/event_plane/test_specific_subscriptions.py
@@ -20,7 +20,8 @@
 from typing import List
 
 import pytest
-from python.event_plane.publisher_subscriber_utils import (
+
+from .event_plane.publisher_subscriber_utils import (
     check_recieved_events,
     gather_published_events,
     run_publishers,