Name:           python-jax
Version:        0.6.0
Release:        %autorelease
# Fill in the actual package summary to submit package to Fedora
Summary:        Differentiate, compile, and transform Numpy code.

# Check if the automatically generated License and its spelling is correct for Fedora
# https://docs.fedoraproject.org/en-US/packaging-guidelines/LicensingGuidelines/
License:        Apache-2.0
URL:            https://github.com/jax-ml/jax
Source:         %{pypi_source jax}

BuildArch:      noarch
BuildRequires:  python3-devel


# Fill in the actual package description to submit package to Fedora
%global _description %{expand:
This is package 'jax' generated automatically by pyp2spec.}

%description %_description

%package -n     python3-jax
Summary:        %{summary}

%description -n python3-jax %_description

# For official Fedora packages, review which extras should be actually packaged
# See: https://docs.fedoraproject.org/en-US/packaging-guidelines/Python/#Extras
%pyproject_extras_subpkg -n python3-jax ci,cpu,cuda,cuda12,cuda12-local,k8s,minimum-jaxlib,rocm,tpu


%prep
%autosetup -p1 -n jax-%{version}


%generate_buildrequires
# Keep only those extras which you actually want to package or use during tests
%pyproject_buildrequires -x ci,cpu,cuda,cuda12,cuda12-local,k8s,minimum-jaxlib,rocm,tpu


%build
%pyproject_wheel


%install
%pyproject_install
# For official Fedora packages, including files with '*' +auto is not allowed
# Replace it with a list of relevant Python modules/globs and list extra files in %%files
%pyproject_save_files '*' +auto


%check
%_pyproject_check_import_allow_no_modules -t


%files -n python3-jax -f %{pyproject_files}


%changelog
%autochangelog