fix : fix imports and renamings after refactoring

Author: abhiprd2000

CHANGELOG.md

@@ -18,7 +18,7 @@ adheres to semantic versioning once it reaches 1.0.
 - **PR #19/20**: Upgraded Rung-3 counterfactuals to use *continuous* vibration RMS outcomes instead of binary labels, enabling direction-sensible physical deltas.
 - Universal `Dataset` contract (`Dataset.from_arrays`) so any vibration dataset
   plugs in without a bespoke loader.
-- Validation scripts for CWRU (`validate_run.py`) and SEU gears
+- Validation scripts for CWRU (`validate_cwru.py`) and SEU gears
   (`validate_seu.py`), and a cross-condition robustness script.
 - Test suite covering the physics, symbolic, causal, consensus, and provider
   layers.

docs/generate_paper_tables.py

@@ -21,7 +21,7 @@
     author='Abhimanyu Prasad, Kazi Tasfin Mahmud',
     url='https://github.com/abhiprd2000/CNSD',
     license='MIT',
-    packages=find_packages(),
+    packages=find_packages(include=['cnsd', 'cnsd.*']),
     python_requires='>=3.11',
     install_requires=['numpy>=2.0', 'scipy>=1.11', 'scikit-learn>=1.4', 'pyyaml>=6.0'],
     extras_require=extras,

examples/cross_rig.py

@@ -0,0 +1 @@
+"""CNSD validation and benchmark scripts (not shipped in the wheel)."""

setup.py

@@ -6,15 +6,15 @@
 import numpy as np
 import scipy.stats as stats
 
-try:
+
+def main():
     os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
     import tensorflow as tf
 
     from cnsd import Dataset
     from cnsd.diagnosis.system import CNSD
     from cnsd.perception.cnn import _train_cnn
 
-    # Argparse
     parser = argparse.ArgumentParser()
     parser.add_argument('--dataset', type=str, required=True, choices=['cwru', 'pu', 'xjtusy'])
     args = parser.parse_args()
@@ -23,7 +23,7 @@
 
     if args.dataset == 'pu':
         from cnsd.physics import PhysicsConfig
-        from validate_pu import load_pu_domain_split
+        from validation.validate_pu import load_pu_domain_split
 
         (X_train_full, y_train_full, cond_train_full), (X_target, y_target, cond_target) = (
             load_pu_domain_split()
@@ -40,7 +40,7 @@
         fs = 64000
 
     elif args.dataset == 'cwru':
-        from validate_run import CWRU, TAXONOMY, load_cwru
+        from validation.validate_cwru import CWRU, TAXONOMY, load_cwru
 
         X, y, cond = load_cwru()
         X = np.asarray(X, np.float32)
@@ -339,7 +339,11 @@ def clone_for_mc(layer):
     print('================ DONE ================')
 
 except Exception:
-    with open('crash_traceback.txt', 'w') as f:
-        traceback.print_exc(file=f)
-    print('CRASHED. Check crash_traceback.txt')
-    sys.exit(1)
+        with open('crash_traceback.txt', 'w') as f:
+            traceback.print_exc(file=f)
+        print('CRASHED. Check crash_traceback.txt')
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    main()

validation/__init__.py

@@ -1,5 +1,5 @@
 """
-validate_run.py - one clean end-to-end validation of the CNSD pipeline on CWRU.
+validate_cwru.py - one clean end-to-end validation of the CNSD pipeline on CWRU.
 
 Purpose: prove the rebuilt system actually RUNS on real data and produces the
 numbers the paper claims - before writing the paper. This is a validation run,
@@ -33,7 +33,7 @@ def load_cwru():
 
     from scipy.io import loadmat
 
-    base_dir = r'E:\301\CWRU-dataset'
+    base_dir = os.environ.get('CNSD_DATA_CWRU', r'E:\301\CWRU-dataset')
     if not os.path.exists(base_dir):
         raise FileNotFoundError(f'CWRU dataset not found at {base_dir}')
 

validation/multi_seed_benchmark.py

@@ -9,12 +9,14 @@
 from cnsd.physics import PhysicsConfig
 
 
-def load_pu_domain_split(data_dir=r'E:\301\PU-dataset', window_size=8192):
+def load_pu_domain_split(data_dir=None, window_size=8192):
     """
     Loads authentic PU dataset and strictly splits by RPM (Domain Shift).
     Train: N09 (900 RPM)
     Test/Calib: N15 (1500 RPM)
     """
+    if data_dir is None:
+        data_dir = os.environ.get('CNSD_DATA_PU', r'E:\301\PU-dataset')
     X_train, y_train, cond_train = [], [], []
     X_target, y_target, cond_target = [], [], []
 

validation/validate_cwru.py

@@ -52,7 +52,7 @@ def load_seu():
     containing 'Data' (~line 16), data starts the next line. Pick one channel
     (pre-commit to it before seeing results - no channel cherry-picking).
     """
-    base_dir = r'E:\301\SEU-dataset\gearbox\gearset'
+    base_dir = os.environ.get('CNSD_DATA_SEU', r'E:\301\SEU-dataset\gearbox\gearset')
 
     label_map = {'Health': 0, 'Chipped': 1, 'Miss': 2, 'Root': 3, 'Surface': 4}
     X_list, y_list, cond_list = [], [], []