diff --git a/src/mflux/generate.py b/src/mflux/generate.py index e4a6640..cbbb3da 100644 --- a/src/mflux/generate.py +++ b/src/mflux/generate.py @@ -3,8 +3,6 @@ import os import sys import time -sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) - from mflux import Flux1, Config, ModelConfig diff --git a/src/mflux/save.py b/src/mflux/save.py index 0bfd4bd..9be9439 100644 --- a/src/mflux/save.py +++ b/src/mflux/save.py @@ -2,8 +2,6 @@ import argparse import os import sys -sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) - from mflux import Flux1, ModelConfig