ModerRAS commited on
Commit
4fd98b7
·
1 Parent(s): 69c4496

Diversify synthetic path episode leaves

Browse files
tools/schema_v2_synthetic_augment/src/main.rs CHANGED
@@ -963,15 +963,21 @@ fn path_seed_compatible_with_base(kind: PathSeedKind, base_kind: BaseMediaKind)
963
  }
964
 
965
  fn simple_path_leaf_records() -> Vec<Record> {
966
- let mut records = vec![
967
- char_record_from_spans(
968
- "03.mkv",
969
- &[(0, 2, Entity::Episode)],
 
 
 
 
970
  "schema_v2_synthetic_path_leaf",
971
- "path_leaf_episode",
972
  "path_leaf_episode",
973
  "path_aug",
974
- ),
 
 
975
  char_record_from_spans(
976
  "file.mkv",
977
  &[],
@@ -988,7 +994,7 @@ fn simple_path_leaf_records() -> Vec<Record> {
988
  "path_leaf_sxe",
989
  "path_aug",
990
  ),
991
- ];
992
 
993
  const MOVIE_TAGS: [&str; 6] = [
994
  "Movie",
 
963
  }
964
 
965
  fn simple_path_leaf_records() -> Vec<Record> {
966
+ let mut records = Vec::new();
967
+ for episode in 1..=120 {
968
+ let episode_text = format!("{episode:02}");
969
+ let filename = format!("{episode_text}.mkv");
970
+ let template_id = format!("path_leaf_episode_{episode_text}");
971
+ records.push(char_record_from_spans(
972
+ &filename,
973
+ &[(0, episode_text.len(), Entity::Episode)],
974
  "schema_v2_synthetic_path_leaf",
975
+ &template_id,
976
  "path_leaf_episode",
977
  "path_aug",
978
+ ));
979
+ }
980
+ records.extend([
981
  char_record_from_spans(
982
  "file.mkv",
983
  &[],
 
994
  "path_leaf_sxe",
995
  "path_aug",
996
  ),
997
+ ]);
998
 
999
  const MOVIE_TAGS: [&str; 6] = [
1000
  "Movie",