fix small error in variable ordering

Julian Kates-Harbeck · Julian Kates-Harbeck · commit 4054c0357175 · 2018-03-24T17:31:37.000-04:00
diff --git a/plasma/preprocessor/normalize.py b/plasma/preprocessor/normalize.py
@@ -141,10 +141,10 @@ def train_on_files(self,shot_files,use_shots,all_machines):
     def cut_end_of_shot(self,shot):
         cut_shot_ends = self.conf['data']['cut_shot_ends']
         if not self.inference_mode and cut_shot_ends: #only cut shots during training
+            T_min_warn = self.conf['data']['T_min_warn']
             if shot.ttd.shape[0] <= T_min_warn:
                 print("not cutting shot since T_min_warn is larger than length of shot")
                 return
-            T_min_warn = self.conf['data']['T_min_warn']
             for key in shot.signals_dict:
                 shot.signals_dict[key] = shot.signals_dict[key][:-T_min_warn,:]
             shot.ttd = shot.ttd[:-T_min_warn]
