diff options
Diffstat (limited to 'lib/ipython_energymodel_prelude.py')
| -rwxr-xr-x | lib/ipython_energymodel_prelude.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/ipython_energymodel_prelude.py b/lib/ipython_energymodel_prelude.py index a44a5d7..6777b17 100755 --- a/lib/ipython_energymodel_prelude.py +++ b/lib/ipython_energymodel_prelude.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 import numpy as np -from dfatool import EnergyModel, RawData, soft_cast_int +from dfatool import PTAModel, RawData, soft_cast_int ignored_trace_indexes = None @@ -10,4 +10,4 @@ files = '../data/20170125_125433_cc1200.tar ../data/20170125_142420_cc1200.tar . raw_data = RawData(files) preprocessed_data = raw_data.get_preprocessed_data() -model = EnergyModel(preprocessed_data, ignore_trace_indexes = ignored_trace_indexes) +model = PTAModel(preprocessed_data, ignore_trace_indexes = ignored_trace_indexes) |