SaiBon99 commited on
Commit
78e9b95
·
1 Parent(s): a7faf87

Add max-batches argument to batch URL scanner for improved processing control

Browse files
src/phising_detection/features/batch_url_scanner.py CHANGED
@@ -670,6 +670,12 @@ def parse_args():
670
  default=None,
671
  help="Number of samples from each input group (default: None = use all)"
672
  )
 
 
 
 
 
 
673
 
674
  return parser.parse_args()
675
 
 
670
  default=None,
671
  help="Number of samples from each input group (default: None = use all)"
672
  )
673
+ parser.add_argument(
674
+ "--max-batches",
675
+ type=int,
676
+ default=None,
677
+ help="Maximum number of batches to process (default: None = process all)"
678
+ )
679
 
680
  return parser.parse_args()
681