kaggle-environments 1.21.0__tar.gz → 1.22.1__tar.gz

{kaggle_environments-1.21.0 → kaggle_environments-1.22.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kaggle-environments
-Version: 1.21.0
+Version: 1.22.1
 Summary: Kaggle Environments
 Author-email: Kaggle <support@kaggle.com>
 Requires-Python: >=3.10

kaggle_environments-1.22.1/docker/Dockerfile

@@ -0,0 +1,34 @@
+# This is set in cloudbuild.yaml, but provide a default
+ARG BASE_IMAGE=gcr.io/kaggle-images/python:latest
+
+# ===== Base Stages =====
+FROM node:22-slim AS node_builder
+FROM ${BASE_IMAGE} AS base
+
+COPY --from=node_builder /usr/local/ /usr/local/
+# Confirm the installation was successful
+RUN node -v && npm -v
+
+WORKDIR /usr/src/app/kaggle_environments
+
+# Copy everything required for the build.
+# This includes the dependency list, the source code, and manifest.
+COPY ./pyproject.toml ./pyproject.toml
+COPY ./MANIFEST.in ./MANIFEST.in
+COPY ./kaggle_environments ./kaggle_environments
+
+# Now that the source code is present, run the installation.
+# This layer is now correctly cached and will only rebuild if the source or pyproject.toml changes.
+RUN uv pip install --system . && \
+    uv cache clean
+
+# 3. Copy any remaining files that aren't part of the build.
+COPY ./README.md ./README.md
+
+# ===== CPU Final Stage =====
+FROM base AS cpu
+CMD ["kaggle-environments"]
+
+# ===== GPU Final Stage =====
+FROM base AS gpu
+CMD ["kaggle-environments"]

{kaggle_environments-1.21.0 → kaggle_environments-1.22.1}/docker/build_cpu.sh

@@ -2,5 +2,5 @@
 path=$(dirname $0)
 # cd to the parent directory to include kaggle_environments folder in Docker build context
 cd $path/..
-docker build --pull -f ./docker/cpu.Dockerfile -t python-simulations-cpu .
+docker build --pull -f ./docker/Dockerfile -t python-simulations-cpu .
 cd -

{kaggle_environments-1.21.0 → kaggle_environments-1.22.1}/docker/build_gpu.sh

@@ -2,5 +2,5 @@
 path=$(dirname $0)
 # cd to the parent directory to include kaggle_environments folder in Docker build context
 cd $path/..
-docker build -f ./docker/gpu.Dockerfile -t python-simulations-gpu .
-cd -
+docker build -f ./docker/Dockerfile --target gpu --build-arg BASE_IMAGE=gcr.io/kaggle-gpu-images/python:latest -t python-simulations-gpu .
+cd -

kaggle_environments-1.22.1/docker/cicd-publish-to-pypi.sh

@@ -0,0 +1,54 @@
+#!/bin/bash
+set -e
+
+# The python:3.11-slim container doesn't have git, which flit needs to build the package.
+echo "Installing git..."
+apt-get update -y && apt-get install -y git --no-install-recommends
+
+# Ensure python dependencies are installed first
+pip install requests --quiet
+
+# Read version directly from pyproject.toml
+VERSION_FROM_PYPROJECT=$(grep -E "^version\s*=\s*\"[0-9]+\.[0-9]+\.[0-9]+\"$" pyproject.toml | cut -d '"' -f 2)
+
+if [ -z "$VERSION_FROM_PYPROJECT" ]; then
+  echo "Could not find version in pyproject.toml. Exiting."
+  exit 1
+fi
+
+echo "Found version $VERSION_FROM_PYPROJECT in pyproject.toml. Checking against PyPI."
+
+cat > /tmp/check_pypi.py << EOL
+import sys
+import requests
+
+package_name = 'kaggle-environments'
+version_to_check = sys.argv[1]
+
+response = requests.get(f'https://pypi.org/pypi/{package_name}/json')
+
+if response.status_code == 200:
+    data = response.json()
+    if version_to_check in data.get('releases', {}):
+        print('true')
+    else:
+        print('false')
+elif response.status_code == 404:
+    print('false')
+else:
+    print(f'Error checking PyPI: {response.status_code}', file=sys.stderr)
+    sys.exit(1)
+EOL
+
+VERSION_EXISTS=$(python3 /tmp/check_pypi.py "$VERSION_FROM_PYPROJECT")
+
+if [ "$VERSION_EXISTS" = "true" ]; then
+  echo "Version $VERSION_FROM_PYPROJECT already exists on PyPI. Skipping publish."
+else
+  echo "Version $VERSION_FROM_PYPROJECT not found on PyPI. Publishing..."
+  pip install flit --quiet
+  export FLIT_USERNAME=__token__
+  # PYPI_TOKEN is passed as an environment variable from cloudbuild
+  export FLIT_PASSWORD=$PYPI_TOKEN
+  flit publish
+fi

kaggle_environments-1.22.1/docker/cloudbuild.yaml

@@ -0,0 +1,68 @@
+steps:
+  - id: "build-cpu-image"
+    name: "gcr.io/cloud-builders/docker"
+    env: ["DOCKER_BUILDKIT=1"]
+    args:
+      [
+        "build",
+        "--file",
+        "docker/Dockerfile",
+        "--target",
+        "cpu",
+        "--tag",
+        "gcr.io/kaggle-images/python-simulations",
+        "--cache-from",
+        "gcr.io/kaggle-images/python-simulations:latest",
+        "--build-arg",
+        "BASE_IMAGE=gcr.io/kaggle-images/python:latest",
+        ".",
+      ]
+    waitFor: ["-"]
+
+  - id: "build-gpu-image"
+    name: "gcr.io/cloud-builders/docker"
+    env: ["DOCKER_BUILDKIT=1"]
+    args:
+      [
+        "build",
+        "--file",
+        "docker/Dockerfile",
+        "--target",
+        "gpu",
+        "--tag",
+        "gcr.io/kaggle-gpu-images/python-simulations",
+        "--cache-from",
+        "gcr.io/kaggle-gpu-images/python-simulations:latest",
+        "--build-arg",
+        "BASE_IMAGE=gcr.io/kaggle-gpu-images/python:latest",
+        ".",
+      ]
+    waitFor: ["-"]
+
+  - id: "test-cpu-image"
+    name: "gcr.io/kaggle-images/python-simulations"
+    waitFor: ["build-cpu-image"]
+    entrypoint: "pytest"
+    args: ["kaggle_environments"]
+
+  - id: "publish-to-pypi"
+    name: "python:3.11-slim"
+    entrypoint: "bash"
+    args: ["docker/cicd-publish-to-pypi.sh"]
+    waitFor: ["test-cpu-image"]
+    secretEnv: ["PYPI_TOKEN"]
+
+images:
+  - "gcr.io/kaggle-images/python-simulations"
+  - "gcr.io/kaggle-gpu-images/python-simulations"
+
+availableSecrets:
+  secretManager:
+    - versionName: projects/464139560241/secrets/pypi-token/versions/latest
+      env: "PYPI_TOKEN"
+
+options:
+  pool:
+    name: "projects/kaggle-cicd/locations/us-east1/workerPools/kaggle-cloud-builders"
+
+timeout: "3600s"

{kaggle_environments-1.21.0 → kaggle_environments-1.22.1}/kaggle_environments/__init__.py

@@ -14,15 +14,27 @@
 
 import os
 from importlib import import_module
+from importlib.metadata import PackageNotFoundError, version
 from os import listdir
+
 from .agent import Agent
-from .api import get_episode_replay, list_episodes, list_episodes_for_team, list_episodes_for_submission
+from .api import (
+    get_episode_replay,
+    list_episodes,
+    list_episodes_for_team,
+    list_episodes_for_submission,
+)
 from .core import evaluate, make, register
 from .main import http_request
 from . import errors
 from . import utils
 
-__version__ = "1.21.0"
+try:
+    __version__ = version("kaggle-environments")
+except PackageNotFoundError:
+    # Package is not installed. This is a fallback for development mode.
+    __version__ = "dev"
+
 
 __all__ = [
     "Agent",