Spaces:
Sleeping
Sleeping
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 |
|