Skip to content

Commit 670df12

Browse files
author
sethsteenken
committed
Event simulator and sample data path fixes
1 parent 310f363 commit 670df12

3 files changed

Lines changed: 9 additions & 4 deletions

File tree

src/entities/asset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from datetime import datetime
33
import random
44
import uuid
5-
from event import Event
5+
from entities.event import Event
66

77
@dataclass
88
class Asset:

src/sample_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ def generate_sample_data(
193193

194194
try:
195195
# Set path as a relative directory in the current src directory
196-
path = "../../infra/data"
196+
path = "../infra/data"
197197

198198
print(f"Generating sample data...")
199199
print(f"Output directory: {os.path.abspath(path)}")

src/simulator/event_simulator.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,14 +52,19 @@
5252
import sys
5353
import threading
5454
import time
55-
from datetime import datetime, timezone
55+
import sys
5656
from pathlib import Path
57+
58+
# Add parent directory to Python path to allow imports from sibling directories
59+
sys.path.insert(0, str(Path(__file__).parent.parent))
60+
61+
from datetime import datetime, timezone
5762
from typing import List, Dict, Optional
5863

5964
# Import our project classes
6065
from entities.event import Event
6166
from entities.asset import AssetType
62-
from event_hub_service import EventHubService
67+
from simulator.event_hub_service import EventHubService
6368
from azd_env_loader import AZDEnvironmentLoader
6469

6570
class AssetSimulator:

0 commit comments

Comments
 (0)