Address commments

Signed-off-by: Adam Li <[email protected]>

Author: adam2392

pywhy_stats/conditional_ksample/base.py renamed to pywhy_stats/conditional_ksample/base_propensity.py

@@ -24,7 +24,7 @@
 )
 
 from ..pvalue_result import PValueResult
-from .base import _compute_propensity_scores, _preprocess_propensity_data, compute_null
+from .base_propensity import _compute_propensity_scores, _preprocess_propensity_data, compute_null
 
 
 def condind(

pywhy_stats/conditional_ksample/bregman.py

@@ -23,7 +23,7 @@
 )
 
 from ..pvalue_result import PValueResult
-from .base import _compute_propensity_scores, _preprocess_propensity_data, compute_null
+from .base_propensity import _compute_propensity_scores, _preprocess_propensity_data, compute_null
 
 
 # XXX: determine if we can do this with Y being optional.

pywhy_stats/conditional_ksample/kcd.py

@@ -2,7 +2,7 @@
 import pytest
 from numpy.testing import assert_array_almost_equal
 
-from pywhy_stats.conditional_ksample.base import compute_null
+from pywhy_stats.conditional_ksample.base_propensity import compute_null
 
 
 # Define a dummy test statistic function for testing

tests/conditional_ksample/test_base.py

@@ -192,7 +192,7 @@ def test_cd_simulation_multi_environment(cd_func, cd_kwargs):
 
     res = cd_func.condind(
         X=df[["x"]],
-        Y=df[["z"]].copy(),
+        Y=df[["z"]],
         group_ind=df[group_col],
         null_sample_size=null_sample_size,
         n_jobs=-1,
@@ -223,7 +223,7 @@ def test_cd_simulation_multi_environment(cd_func, cd_kwargs):
 
     res = cd_func.condind(
         X=df[["x", "x1"]],
-        Y=df[["z"]].copy(),
+        Y=df[["z"]],
         group_ind=df[group_col],
         null_sample_size=null_sample_size,
         n_jobs=-1,