Update main_pc_apply_drift.py

Solves #7 confusion
HilaManor · Aug 14, 2024 · c369ad3 · c369ad3
1 parent a057f63
commit c369ad3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/code/main_pc_apply_drift.py b/code/main_pc_apply_drift.py
@@ -36,6 +36,10 @@
     args = parser.parse_args()
     args.shift_x0_for_np = True
     args.sub_iters = None
+
+    # Input check
+    if args.drift_start < args.drift_end:
+        raise ValueError('Drift start must be greater than drift end')
 
     set_reproducability(args.seed)