update assignment.py to use relative imports

amplitude · Aug 18, 2023 · 9e6976b · 9e6976b
1 parent f7b25bb
commit 9e6976b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/amplitude_experiment/assignment/assignment.py b/src/amplitude_experiment/assignment/assignment.py
@@ -1,8 +1,8 @@
 import time
 from typing import Dict
 
-from src.amplitude_experiment.flagresult import FlagResult
-from src.amplitude_experiment.user import User
+from ..flagresult import FlagResult
+from ..user import User
 
 DAY_MILLIS = 24 * 60 * 60 * 1000