File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11import os
22import sys
3-
4- # Ensure CausalTune is in the Python path
5- root_path = os .path .realpath ("../../../.." )
6- sys .path .append (os .path .join (root_path , "causaltune" ))
7-
83import pickle
94import numpy as np
105import matplotlib
1712
1813warnings .filterwarnings ("ignore" )
1914
20- from causaltune import CausalTune
21- from causaltune .datasets import (
22- load_dataset ,
23- )
24- from causaltune .models .passthrough import passthrough_model
15+ # Ensure CausalTune is in the Python path
16+ root_path = os .path .realpath ("../../../.." )
17+ sys .path .append (os .path .join (root_path , "causaltune" )) # noqa: E402
18+
19+ # Import CausalTune and other custom modules after setting up the path
20+ from causaltune import CausalTune # noqa: E402
21+ from causaltune .datasets import load_dataset # noqa: E402
22+ from causaltune .models .passthrough import passthrough_model # noqa: E402
2523
2624
2725def parse_arguments ():
You can’t perform that action at this time.
0 commit comments