1{
2 lib,
3 buildPythonPackage,
4 fetchPypi,
5}:
6
7buildPythonPackage rec {
8 pname = "inference-gym";
9 version = "0.0.5";
10 format = "wheel";
11
12 src = fetchPypi {
13 inherit format version;
14 pname = "inference_gym";
15 dist = "py3";
16 python = "py3";
17 hash = "sha256-E3lNgCZIObPBkl0PWACUG19XOiCOh1+N/sUFHQyA/wE=";
18 };
19
20 pythonImportsCheck = [ "inference_gym" ];
21
22 # The package does not ship any test.
23 doCheck = false;
24
25 meta = {
26 description = "Place to exercise inference methods to help make them faster, leaner and more robust";
27 homepage = "https://github.com/tensorflow/probability/tree/main/spinoffs/inference_gym";
28 license = lib.licenses.asl20;
29 maintainers = with lib.maintainers; [ GaetanLepage ];
30 };
31}