Adil Zouitine a7d1179aab fix(processor): Preserve stats overrides in normalizer load_state_dict and fix training resumption (#1958) ...

* feat(processor): enhance normalization handling and state management

- Added support for additional normalization modes including IDENTITY.
- Introduced a new function `clean_state_dict` to remove specific substrings from state dict keys.
- Implemented preservation of explicitly provided normalization statistics during state loading.
- Updated training script to conditionally provide dataset statistics based on resume state.
- Expanded tests to verify the correct behavior of stats override preservation and loading.

* fix(train): remove redundant comment regarding state loading

- Removed a comment that noted the preprocessor and postprocessor state is already loaded when resuming training, as it was deemed unnecessary for clarity.

2025-09-16 16:45:13 +02:00

..

test_act_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_batch_conversion.py

refactor(processor): clarify action types, distinguish PolicyAction, RobotAction, and EnvAction (#1908)

2025-09-10 22:40:37 +02:00

test_batch_processor.py

test(processor): fix batch expectation

2025-09-15 22:22:17 +02:00

test_classifier_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_converters.py

refactor(processors): several additions (#1926)

2025-09-13 23:53:20 +02:00

test_device_processor.py

test(processor): all processors use now the same create_transition (#1906)

2025-09-10 18:39:06 +02:00

test_diffusion_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_normalize_processor.py

fix(processor): Preserve stats overrides in normalizer load_state_dict and fix training resumption (#1958)

2025-09-16 16:45:13 +02:00

test_observation_processor.py

fix(processor): specialized processors respect contract by raising if none (#1909)

2025-09-10 18:45:47 +02:00

test_pi0_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_pipeline.py

refactor(processor): replace ModelHubMixin with HubMixin and enhance save_pretrained method (#1937)

2025-09-15 13:13:35 +02:00

test_rename_processor.py

fix(processor): specialized processors respect contract by raising if none (#1909)

2025-09-10 18:45:47 +02:00

test_sac_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_smolvla_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_tdmpc_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00

test_tokenizer_processor.py

fix(processor): specialized processors respect contract by raising if none (#1909)

2025-09-10 18:45:47 +02:00

test_vqbet_processor.py

feat(processor): enhance type safety with generic DataProcessorPipeline for policy and robot pipelines (#1915)

2025-09-11 13:36:04 +02:00