diff --git a/icp/python/src/triton_distributed/icp/event_plane.py b/icp/python/src/triton_distributed/icp/event_plane.py index f4dab768..17ee4aa7 100644 --- 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 index 368ee016..7d2c7d32 100644 --- 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 index 9d53e43a..4ea6f55b 100644 --- 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 index d33e7960..cabd95cd 100644 --- 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 index b18c586d..85d176a6 100644 --- 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,