Skip to content

Commit 638ba1d

Browse files
Merge pull request #6 from dimitra-maoutsa/testing
Update DeterministicParticleFlowControl.py imports
2 parents 676c4a6 + 5c4a384 commit 638ba1d

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

DeterministicParticleFlowControl/DeterministicParticleFlowControl.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,12 @@
1717
from matplotlib import pyplot as plt
1818
#import score_estimators
1919
#import reweighting
20-
from score_estimators.score_function_estimators import score_function_multid_seperate
21-
#from score_estimators import score_function_estimators.score_function_multid_seperate
22-
#from score import score_function_multid_seperate2
23-
from reweighting.optimal_transport_reweighting import reweight_optimal_transport_multidim
20+
if __name__ == "DeterministicParticleFlowControl.DeterministicParticleFlowControl":
21+
from .score_estimators.score_function_estimators import score_function_multid_seperate
22+
from .reweighting.optimal_transport_reweighting import reweight_optimal_transport_multidim
23+
else:
24+
from score_estimators.score_function_estimators import score_function_multid_seperate
25+
from reweighting.optimal_transport_reweighting import reweight_optimal_transport_multidim
2426
from duecredit import due, BibTeX
2527

2628

0 commit comments

Comments
 (0)