Merge branch '372' into 431

HealthCatalyst · Nov 10, 2017 · 697bc12 · 697bc12
2 parents ec610c8 + 12e19b0
commit 697bc12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/healthcareai/tests/test_feature_availability_profiler.py b/healthcareai/tests/test_feature_availability_profiler.py
@@ -20,7 +20,7 @@ def setUp(self):
         for i in range(1000):
             random_second = randrange(int_delta)
             admit[i] = pd.datetime(2015, 5, 1) + timedelta(seconds=random_second)
-        self.df['AdmitDTS'] = admit
+        self.df['AdmitDTS'] = admit.astype(np.datetime64)
         # add nulls
         a = np.random.rand(1000) > .5
         self.df.loc[a, ['A']] = np.nan
@@ -91,7 +91,7 @@ def setUp(self):
             random_second = randrange(int_delta)
             admit[i] = pd.datetime(2015, 5, 1) + timedelta(
                 seconds=random_second)
-        self.df['AdmitDTS'] = admit
+        self.df['AdmitDTS'] = admit.astype(np.datetime64)
 
     def runTest(self):
         with self.assertRaises(HealthcareAIError) as error: