(actual_result: bytes, golden_result: bytes)
| 501 | gc.collect() |
| 502 | |
| 503 | def comparison_fn(actual_result: bytes, golden_result: bytes): |
| 504 | actual_run_values = pickle.loads(actual_result) |
| 505 | golden_run_values = pickle.loads(golden_result) |
| 506 | self.assertNestedEqual( |
| 507 | actual_run_values["inputs"], golden_run_values["inputs"] |
| 508 | ) |
| 509 | self.assertNestedAllClose( |
| 510 | actual_run_values["trainer_state"], |
| 511 | golden_run_values["trainer_state"], |
| 512 | rtol=5e-6, |
| 513 | atol=0, |
| 514 | ) |
| 515 | self.assertNestedEqual( |
| 516 | actual_run_values["trainer_state_specs"], |
| 517 | golden_run_values["trainer_state_specs"], |
| 518 | ) |
| 519 | |
| 520 | return result, comparison_fn |
| 521 |
nothing calls this directly
no test coverage detected