| from inference.core.active_learning.entities import ( | |
| ActiveLearningConfiguration, | |
| BatchReCreationInterval, | |
| ) | |
| from inference.core.active_learning.utils import ( | |
| generate_start_timestamp_for_this_month, | |
| generate_start_timestamp_for_this_week, | |
| generate_today_timestamp, | |
| ) | |
| RECREATION_INTERVAL2TIMESTAMP_GENERATOR = { | |
| BatchReCreationInterval.DAILY: generate_today_timestamp, | |
| BatchReCreationInterval.WEEKLY: generate_start_timestamp_for_this_week, | |
| BatchReCreationInterval.MONTHLY: generate_start_timestamp_for_this_month, | |
| } | |
| def generate_batch_name(configuration: ActiveLearningConfiguration) -> str: | |
| batch_name = configuration.batches_name_prefix | |
| if configuration.batch_recreation_interval is BatchReCreationInterval.NEVER: | |
| return batch_name | |
| timestamp_generator = RECREATION_INTERVAL2TIMESTAMP_GENERATOR[ | |
| configuration.batch_recreation_interval | |
| ] | |
| timestamp = timestamp_generator() | |
| return f"{batch_name}_{timestamp}" | |