Merge branch 'gbm_work_7' into gbm_work_8

google · Mar 28, 2024 · c3016ac · c3016ac
2 parents 26950f2 + e77545f
commit c3016ac
Show file tree

Hide file tree

Showing 26 changed files with 2,285 additions and 1,407 deletions.
diff --git a/.github/workflows/coverage.yaml b/.github/workflows/coverage.yaml
@@ -1,12 +1,6 @@
 name: coverage
 
-on:
-  # this allows for `contents: write` and `pull-requests: write` from forks
-  pull_request_target:
-
-# limit default permissions to just read-only checkouts
-permissions:
-  contents: read
+on: [push, pull_request]
 
 jobs:
   coverage:
@@ -42,7 +36,7 @@ jobs:
           cache: poetry
 
       - name: Install dependencies
-        run: poetry install --no-interaction
+        run: poetry install --no-interaction --extras polars
 
       - name: Run coverage
         id: coverage
@@ -68,14 +62,4 @@ jobs:
             const diff = pr - main;
             const report = `### Coverage report
             **Main**: ${main.toFixed(2)}% | **PR**: ${pr.toFixed(2)}% | **Diff: ${diff.toFixed(2)} ${diff >= 0 ? '✅' : '⚠️'}**`;
-            await core.summary.addRaw(report).write()
-            try {
-              github.rest.issues.createComment({
-                issue_number: context.issue.number,
-                owner: context.repo.owner,
-                repo: context.repo.repo,
-                body: report
-              })
-            } catch (e) {
-              console.log("Could not post comment to pull request", e)
-            }
+            await core.summary.addRaw(report, true).write();
diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
@@ -17,7 +17,7 @@ jobs:
         python-version: ["3.8", "3.9", "3.10", "3.11"]
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v3  # v4 doesn't work on the container
 
       - name: Build package
         run: |
@@ -34,10 +34,16 @@ jobs:
   # macos-XXX-large are Intel runners, macos-XXX-xlarge are arm64 runners
   # https://docs.github.com/en/actions/using-github-hosted-runners/about-larger-runners/about-larger-runners
   build-macos-intel:
-    runs-on: macos-13-large
+    runs-on: ${{ matrix.platform.runner }}
     strategy:
       matrix:
         python-version: ["3.8", "3.9", "3.10", "3.11"]
+        platform:
+          [
+            { runner: macos-12-large, artifact: macos12-intel },
+            { runner: macos-13-large, artifact: macos13-intel },
+            { runner: macos-14-large, artifact: macos14-intel },
+          ]
     steps:
       - name: Checkout
         uses: actions/checkout@v4
@@ -56,14 +62,20 @@ jobs:
       - name: Save package for following jobs
         uses: actions/upload-artifact@v3
         with:
-          name: dist-macos-intel-${{ matrix.python-version }}
+          name: dist-${{ matrix.platform.artifact }}-${{ matrix.python-version }}
           path: dist
 
   build-macos-arm:
-    runs-on: macos-13-xlarge
+    runs-on: ${{ matrix.platform.runner }}
     strategy:
       matrix:
         python-version: ["3.8", "3.9", "3.10", "3.11"]
+        platform:
+          [
+            # there is not runner for macos12 on arm
+            { runner: macos-13-xlarge, artifact: macos-13-arm },
+            { runner: macos-14-xlarge, artifact: macos-14-arm }
+          ]
     steps:
       - name: Checkout
         uses: actions/checkout@v4
@@ -90,7 +102,7 @@ jobs:
       - name: Save package for following jobs
         uses: actions/upload-artifact@v3
         with:
-          name: dist-macos-arm-${{ matrix.python-version }}
+          name: dist-${{ matrix.platform.artifact }}-${{ matrix.python-version }}
           path: dist
 
   test-install:
@@ -101,7 +113,9 @@ jobs:
         platform:
           [
             { runner: ubuntu-latest, artifact: ubuntu },
-            { runner: macos-13-large, artifact: macos-intel },
+            { runner: macos-12-large, artifact: macos12-intel },
+            { runner: macos-13-large, artifact: macos13-intel },
+            { runner: macos-14-large, artifact: macos14-intel },
           ]
     runs-on: ${{ matrix.platform.runner }}
     env:
@@ -110,7 +124,7 @@ jobs:
       SYSTEM_VERSION_COMPAT: 0
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v3
 
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v4
@@ -131,13 +145,18 @@ jobs:
 
   test-macos-arm-install:
     needs: [build-macos-arm]
-    runs-on: macos-13-xlarge
     strategy:
       matrix:
         python-version: ["3.8", "3.9", "3.10", "3.11"]
+        platform:
+          [
+            { runner: macos-13-xlarge, artifact: macos-13-arm },
+            { runner: macos-14-xlarge, artifact: macos-14-arm }
+          ]
+    runs-on: ${{ matrix.platform.runner }}
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v3
 
       - name: Install Python
         run: |
@@ -152,7 +171,7 @@ jobs:
       - name: Download build
         uses: actions/download-artifact@v3
         with:
-          name: dist-macos-arm-${{ matrix.python-version }}
+          name: dist-${{ matrix.platform.artifact }}-${{ matrix.python-version }}
           path: dist
 
       - name: Install from wheel
@@ -169,7 +188,7 @@ jobs:
       POETRY_PYPI_TOKEN_PYPI: ${{ secrets.PYPI_TOKEN }}
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v3
 
       - name: Install poetry
         run: pip install poetry

diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -35,7 +35,7 @@ jobs:
           cache: poetry
 
       - name: Install dependencies
-        run: poetry install --no-interaction
+        run: poetry install --no-interaction --extras polars
 
       - name: Run tests
         run: poetry run bazel test //temporian/...:all //docs/...:all //tools/...:all --test_output=errors --compilation_mode=dbg --config=linux

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -8,6 +8,10 @@
 
 ### Fixes
 
+## 0.8.1
+
+Adding missing wheels for macos, no changes to the library
+
 ## 0.8.0
 
 ### Features

diff --git a/docs/public_api_test.py b/docs/public_api_test.py
@@ -39,8 +39,10 @@
     "to_csv",
     "from_csv",
     "to_pandas",
+    "to_polars",
     "to_numpy",
     "from_pandas",
+    "from_polars",
     "to_parquet",
     "from_parquet",
     "to_tensorflow_dataset",

diff --git a/docs/src/reference/index.md b/docs/src/reference/index.md
@@ -29,6 +29,8 @@ Check the index on the left for a more detailed description of any symbol.
 | ------------------------------------------- | --------------------------------------------------------------------- |
 | [`tp.from_pandas()`][temporian.from_pandas] | Converts a Pandas DataFrame into an [`EventSet`][temporian.EventSet]. |
 | [`tp.to_pandas()`][temporian.to_pandas]     | Converts an [`EventSet`][temporian.EventSet] to a pandas DataFrame.   |
+| [`tp.from_polars()`][temporian.from_polars] | Converts a Polars DataFrame into an [`EventSet`][temporian.EventSet]. |
+| [`tp.to_polars()`][temporian.to_polars]     | Converts an [`EventSet`][temporian.EventSet] to a polars DataFrame.   |
 | [`tp.from_csv()`][temporian.from_csv]       | Reads an [`EventSet`][temporian.EventSet] from a CSV file.            |
 | [`tp.to_csv()`][temporian.to_csv]           | Saves an [`EventSet`][temporian.EventSet] to a CSV file.              |
 

diff --git a/docs/src/reference/temporian/io/from_polars.md b/docs/src/reference/temporian/io/from_polars.md
diff --git a/docs/src/reference/temporian/io/to_polars.md b/docs/src/reference/temporian/io/to_polars.md