Skip to content

Conversation

@mlipbot
Copy link
Collaborator

@mlipbot mlipbot commented Jan 6, 2026

External release PR to publish new release.

@github-actions
Copy link
Contributor

github-actions bot commented Jan 6, 2026

Coverage

Coverage Report
FileStmtsMissCoverMissing
src/mlip
   __init__.py20100% 
src/mlip/data
   __init__.py80100% 
   chemical_system.py43588%74, 77, 80, 89, 97
   configs.py42392%84–86
   dataset_info.py34294%59, 62
   graph_dataset_builder.py1501093%111, 193, 235, 239, 349–350, 366, 370, 380–381
src/mlip/data/chemical_systems_readers
   __init__.py40100% 
   chemical_systems_reader.py16193%88
   combined_reader.py160100% 
   extxyz_reader.py711184%101, 108–111, 117, 126, 132–135
   hdf5_reader.py48687%81, 86–89, 119
   type_aliases.py40100% 
   utils.py340100% 
src/mlip/data/helpers
   __init__.py50100% 
   atomic_energies.py23386%56–57, 61
   atomic_number_table.py23195%49
   data_prefetching.py52590%109, 140, 146–148
   data_split.py69297%188, 191
   dummy_init_graph.py80100% 
   dynamically_batch.py51786%108, 121–122, 125–128
   edge_vectors.py140100% 
   graph_creation.py150100% 
   graph_dataset.py661774%97–98, 102, 155–163, 165, 167–168, 180–181
   neighbor_analysis.py280100% 
   neighborhood.py26196%37
src/mlip/inference
   __init__.py10100% 
   batched_inference.py73297%56, 171
src/mlip/models
   __init__.py50100% 
   atomic_energies.py20860%60, 72–76, 80, 86
   blocks.py82495%53, 56, 147, 153
   force_field.py420100% 
   loss.py63887%51, 84, 88, 92, 145–146, 196, 209
   loss_helpers.py1242877%48, 52, 64, 115–116, 126–127, 141–142, 157–158, 174–175, 177–179, 185–186, 188–190, 250–251, 253, 259, 308–310
   mlip_network.py15286%51, 62
   model_io.py370100% 
   options.py37391%106, 109–110
   params_loading.py36877%44, 76, 90–92, 95, 98, 103
   params_transfer.py18194%32
   predictor.py62198%63
   radial_embedding.py120100% 
src/mlip/models/mace
   blocks.py75297%170, 173
   config.py37391%109, 113, 122
   message_passing.py270100% 
   models.py170994%98, 100, 202, 215, 352, 418–420, 422
   symmetric_contraction.py58296%43, 79
src/mlip/models/nequip
   config.py170100% 
   models.py151298%153, 282
   nequip_helpers.py42490%35–37, 63
src/mlip/models/visnet
   blocks.py1973084%47, 50, 78–80, 83, 85–87, 90–92, 95–97, 125, 211, 214–217, 235, 238–239, 242, 251, 310, 314, 326, 343
   config.py160100% 
   models.py1590100% 
src/mlip/simulation
   __init__.py20100% 
   enums.py110100% 
   exceptions.py20100% 
   simulation_engine.py29389%76, 83, 102
   state.py130100% 
   temperature_scheduling.py20480%36, 48, 76–77
   utils.py23291%66, 74
src/mlip/simulation/ase
   __init__.py10100% 
   ase_simulation_engine.py129695%68, 106–107, 145, 232, 256
   mlip_ase_calculator.py61985%135, 137, 142, 165–166, 170, 173, 178–179
src/mlip/simulation/configs
   __init__.py30100% 
   ase_config.py28292%90, 96
   jax_md_config.py28389%75–76, 78
   simulation_config.py220100% 
src/mlip/simulation/jax_md
   __init__.py10100% 
   helpers.py87198%58
   jax_md_simulation_engine.py1952288%189, 195–197, 201, 221, 226–228, 234, 242–243, 256–257, 262, 266, 603–605, 608, 613, 616
   jaxmd_utils.py630100% 
   states.py190100% 
src/mlip/training
   __init__.py60100% 
   ema.py220100% 
   evaluation.py44393%55, 87, 124
   finetuning_utils.py22220%15, 17–19, 21–22, 24, 27, 30–31, 35, 38–40, 42–43, 45, 48, 65–66, 69, 73
   metrics_reweighting.py80100% 
   optimizer.py241729%25, 27–28, 33–34, 38, 40, 66–67, 71, 73, 135–136, 138, 140, 143, 145
   optimizer_config.py160100% 
   training_io_handler.py1333573%139, 188, 197, 214, 225, 231–233, 252–253, 257–260, 262–264, 271–273, 275–276, 280, 284, 290, 294, 298, 326, 331, 335–336, 338, 343, 347, 351
   training_loggers.py604721%42–43, 45, 54, 59–60, 62–63, 65–68, 70, 87, 89–95, 98–99, 103–104, 108–113, 115–117, 132, 134–135, 141, 143–145, 147–148, 150–151, 153–154
   training_loop.py1812387%116–117, 163, 184, 225, 329–331, 337–340, 344, 348, 359, 376–379, 406–408, 427
   training_loop_config.py130100% 
   training_state.py340100% 
   training_step.py44393%58, 78, 149
src/mlip/typing
   __init__.py40100% 
   fields.py100100% 
   graph_definition.py230100% 
   prediction.py80100% 
   signatures.py70100% 
src/mlip/utils
   __init__.py00100% 
   dict_flatten.py40685%73, 85–87, 121, 123
   jax_utils.py10100% 
   multihost.py332136%52–53, 55–58, 60, 62, 64, 112–113, 115, 118, 134–137, 140–143
   no_pbc_cell.py50100% 
   safe_norm.py50100% 
TOTAL390842089% 

Tests Skipped Failures Errors Time
68 0 💤 0 ❌ 0 🔥 4m 54s ⏱️

@chrbrunk chrbrunk merged commit 27265a5 into main Jan 6, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants