v0.4.1

oumi-ai/oumiv0.4.1Oct 14, 2025by shanghongsim

AI Summary

Patch release with bug fixes including NaN handling in dataset analyzer, SLURM environment variable fixes, and various infrastructure improvements.

Key Highlights

  • Fixed NaN values in dataset analyzer
  • Fixed SLURM environment variables (PMI_RANK → SLURM_PROCID)
  • SkyPilot compatibility fixes
  • Unique inference scratch filenames via hashing

New Features

  • Bug fixes for dataset analyzer
  • SLURM environment variable fixes
  • SkyPilot unit test fixes
  • GPU test improvements
  • Optional stdout_file in SLURM client
  • Lazy cloud initialization
  • Oumi dataset support in synthesis

Full Release Notes

## What's Changed
* Fix NaN values in dataset analyzer statistics for single conversations by @ryan-arman in https://github.com/oumi-ai/oumi/pull/1961
* [tiny] Add `__init__.py` to some test dirs by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1963
* Add The Ability To Trail Logs For Launcher Jobs by @rlehman221 in https://github.com/oumi-ai/oumi/pull/1951
* Move compute statistics to analysis_utils by @ryan-arman in https://github.com/oumi-ai/oumi/pull/1962
* Fixed Sky Pilot Unit Tests Failing by @rlehman221 in https://github.com/oumi-ai/oumi/pull/1967
* extract conversation_turns from conversation_level_summary to top level by @ryan-arman in https://github.com/oumi-ai/oumi/pull/1969
* Update README.md by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1968
* [tiny] Upgrade transformers to 4.56 by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1966
* Quick fix for our GPU tests by @taenin in https://github.com/oumi-ai/oumi/pull/1970
* Support NERSC Perlmutter HPC cluster in Oumi launcher by @wizeng23 in https://github.com/oumi-ai/oumi/pull/1959
* Added Launcher Logs Command by @rlehman221 in https://github.com/oumi-ai/oumi/pull/1964
* Make stdout_file optional in slurm client by @rlehman221 in https://github.com/oumi-ai/oumi/pull/1974
* Add synth documentation and example configs by @jgreer013 in https://github.com/oumi-ai/oumi/pull/1965
* Add cuda launch blocking arg to e2e tests for debugging by @jgreer013 in https://github.com/oumi-ai/oumi/pull/1976
* Pin uvicorn version to fix skypilot by @jgreer013 in https://github.com/oumi-ai/oumi/pull/1978
* Update inference to always hash for unique inference scratch filenames by @jgreer013 in https://github.com/oumi-ai/oumi/pull/1986
* Lazy init clouds in oumi launcher by @oelachqar in https://github.com/oumi-ai/oumi/pull/1985
* Unblock oumi dataset support in synthesis  by @jgreer013 in https://github.com/oumi-ai/oumi/pull/1988


**Full Changelog**: https://github.com/oumi-ai/oumi/compare/v0.4.0...v0.4.1