Fixed bugs in classify.py when classifying redshift.

daniel-muthukrishna · Jun 1, 2018 · 117dd61 · 117dd61
1 parent 6975c80
commit 117dd61
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dash/classify.py b/dash/classify.py
@@ -94,7 +94,7 @@ def list_best_matches(self, n=5, saveFilename='DASH_matches.txt'):
             for i in range(20):
                 host, name, age = classification_split(bestTypes[specNum][i])
                 if not self.knownZ:
-                    redshifts.append(self.calc_redshift(inputImages[i], name, age)[0], inputMinMaxIndexes[i])
+                    redshifts.append(self.calc_redshift(inputImages[i], name, age, inputMinMaxIndexes[i])[0])
                 prob = softmaxes[specNum][i]
                 bestMatchList.append((host, name, age, prob))
             bestMatchList = np.array(bestMatchList)
@@ -162,7 +162,6 @@ def calc_redshift(self, inputFlux, snName, snAge, inputMinMaxIndex):
         templateFluxes = []
         templateMinMaxIndexes = []
         for i in range(numOfSubTemplates):
-            templateNames = []
             templateFluxes.append(snInfos[i][1])
             templateMinMaxIndexes.append((snInfos[i][2], snInfos[i][3]))