diff-diff 2.1.0__tar.gz → 2.1.2__tar.gz

{diff_diff-2.1.0 → diff_diff-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diff-diff
-Version: 2.1.0
+Version: 2.1.2
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Science/Research
 Classifier: Operating System :: OS Independent

{diff_diff-2.1.0 → diff_diff-2.1.2}/diff_diff/__init__.py

@@ -71,6 +71,11 @@ from diff_diff.prep import (
     balance_panel,
     create_event_time,
     generate_did_data,
+    generate_ddd_data,
+    generate_event_study_data,
+    generate_factor_data,
+    generate_panel_data,
+    generate_staggered_data,
     make_post_indicator,
     make_treatment_indicator,
     rank_control_units,
@@ -131,7 +136,7 @@ from diff_diff.datasets import (
     load_mpdta,
 )
 
-__version__ = "2.1.0"
+__version__ = "2.1.2"
 __all__ = [
     # Estimators
     "DifferenceInDifferences",
@@ -190,6 +195,11 @@ __all__ = [
     "validate_did_data",
     "summarize_did_data",
     "generate_did_data",
+    "generate_staggered_data",
+    "generate_factor_data",
+    "generate_ddd_data",
+    "generate_panel_data",
+    "generate_event_study_data",
     "create_event_time",
     "aggregate_to_cohorts",
     "rank_control_units",

{diff_diff-2.1.0 → diff_diff-2.1.2}/diff_diff/_backend.py

@@ -23,6 +23,10 @@ try:
         project_simplex as _rust_project_simplex,
         solve_ols as _rust_solve_ols,
         compute_robust_vcov as _rust_compute_robust_vcov,
+        # TROP estimator acceleration
+        compute_unit_distance_matrix as _rust_unit_distance_matrix,
+        loocv_grid_search as _rust_loocv_grid_search,
+        bootstrap_trop_variance as _rust_bootstrap_trop_variance,
     )
     _rust_available = True
 except ImportError:
@@ -32,6 +36,10 @@ except ImportError:
     _rust_project_simplex = None
     _rust_solve_ols = None
     _rust_compute_robust_vcov = None
+    # TROP estimator acceleration
+    _rust_unit_distance_matrix = None
+    _rust_loocv_grid_search = None
+    _rust_bootstrap_trop_variance = None
 
 # Determine final backend based on environment variable and availability
 if _backend_env == 'python':
@@ -42,6 +50,10 @@ if _backend_env == 'python':
     _rust_project_simplex = None
     _rust_solve_ols = None
     _rust_compute_robust_vcov = None
+    # TROP estimator acceleration
+    _rust_unit_distance_matrix = None
+    _rust_loocv_grid_search = None
+    _rust_bootstrap_trop_variance = None
 elif _backend_env == 'rust':
     # Force Rust mode - fail if not available
     if not _rust_available:
@@ -61,4 +73,8 @@ __all__ = [
     '_rust_project_simplex',
     '_rust_solve_ols',
     '_rust_compute_robust_vcov',
+    # TROP estimator acceleration
+    '_rust_unit_distance_matrix',
+    '_rust_loocv_grid_search',
+    '_rust_bootstrap_trop_variance',
 ]