Merge pull request #63 from sdfordham/placebo-test-time-period

Remove coerce to integers in PlaceboTest
sdfordham · May 20, 2024 · 81c9abe · 81c9abe
2 parents 61cc0dc + c0f1bbd
commit 81c9abe
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pysyncon/utils.py b/pysyncon/utils.py
@@ -246,10 +246,9 @@ def _single_placebo(
         """
         scm.fit(dataprep=dataprep, **scm_options)
 
-        min_ = int(min(dataprep.foo[dataprep.time_variable]))
-        max_ = int(max(dataprep.foo[dataprep.time_variable]))
-
-        Z0, Z1 = dataprep.make_outcome_mats(time_period=range(min_, max_))
+        Z0, Z1 = dataprep.make_outcome_mats(
+            time_period=dataprep.foo[dataprep.time_variable]
+        )
         synthetic = scm._synthetic(Z0=Z0)
         gaps = scm._gaps(Z0=Z0, Z1=Z1)
         return synthetic.rename(dataprep.treatment_identifier), gaps.rename(