Bug fixes

apache · Aug 2, 2015 · b1d009f · b1d009f
1 parent f851d20
commit b1d009f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app/models.py b/app/models.py
@@ -126,6 +126,7 @@ def sync_to_db(cls, name, cluster):
                 col_obj.filterable = True
             if col_obj:
                 col_obj.type = cols[col]['type']
+            col_obj.datasource = datasource
             col_obj.generate_metrics()
         #session.commit()
 
@@ -199,7 +200,7 @@ def generate_metrics(self):
             json=json.dumps({'type': 'count', 'name': 'count'})
         ))
         # Somehow we need to reassign this for UDAFs
-        corrected_type = 'DOUBLE' if self.type in ('DOUBLE', 'FLOAT') else 'self.type'
+        corrected_type = 'DOUBLE' if self.type in ('DOUBLE', 'FLOAT') else self.type
 
         if self.sum and self.isnum:
             mt = corrected_type.lower() + 'Sum'