|
1 | 1 | import shutil |
| 2 | +import warnings |
2 | 3 | from functools import partial |
3 | 4 | from pathlib import Path |
4 | 5 | from typing import TYPE_CHECKING, List |
@@ -431,13 +432,18 @@ def check_ready(self): |
431 | 432 | * False and displays a warning if not |
432 | 433 |
|
433 | 434 | """ |
434 | | - if ( |
435 | | - self.images_filepaths == [] |
436 | | - or self.labels_filepaths == [] |
437 | | - or len(self.images_filepaths) != len(self.labels_filepaths) |
438 | | - ): |
439 | | - logger.warning("Image and label paths are not correctly set") |
440 | | - return False |
| 435 | + if not self.unsupervised_mode: |
| 436 | + if ( |
| 437 | + self.images_filepaths == [] |
| 438 | + or self.labels_filepaths == [] |
| 439 | + or len(self.images_filepaths) != len(self.labels_filepaths) |
| 440 | + ): |
| 441 | + logger.warning("Image and label paths are not correctly set") |
| 442 | + return False |
| 443 | + else: |
| 444 | + if self.get_unsupervised_image_filepaths() == []: |
| 445 | + logger.warning("Image paths are not correctly set") |
| 446 | + return False |
441 | 447 | return True |
442 | 448 |
|
443 | 449 | def _toggle_unsupervised_mode(self, enabled=False): |
@@ -940,8 +946,9 @@ def start(self): |
940 | 946 |
|
941 | 947 | if not self.check_ready(): # issues a warning if not ready |
942 | 948 | err = "Aborting, please set all required paths" |
943 | | - self.log.print_and_log(err) |
| 949 | + # self.log.print_and_log(err) |
944 | 950 | logger.warning(err) |
| 951 | + warnings.warn(err, stacklevel=1) |
945 | 952 | return |
946 | 953 |
|
947 | 954 | if self.worker is not None: |
|
0 commit comments