Remove with matrix.python-version

Author: lresende

.github/workflows/build.yml

@@ -21,8 +21,8 @@ jobs:
         with:
           clean: true
       - uses: jupyterlab/maintainer-tools/.github/actions/base-setup@v1
-        with:
-          python_version: "${{ matrix.python-version }}"
+#        with:
+#          python_version: "${{ matrix.python-version }}"
       - name: Display dependency info
         run: |
           python --version
@@ -33,6 +33,7 @@ jobs:
       - name: Install Python dependencies
         run: |
           pip install ".[test]"
+          pip install --upgrade cloudpickle
       - name: Build and install Jupyter Enterprise Gateway
         uses: nick-invision/[email protected]
         with:

enterprise_gateway/itests/test_python_kernel.py

@@ -122,6 +122,7 @@ def test_run_pi_example(self):
         pi_code = []
         pi_code.append("from random import random\n")
         pi_code.append("from operator import add\n")
+        pi_code.append("from pyspark.cloudpickle import cloudpickle\n")
         pi_code.append("partitions = 20\n")
         pi_code.append("n = 100000 * partitions\n")
         # Define the function with explicit imports inside to avoid serialization issues
@@ -130,6 +131,8 @@ def test_run_pi_example(self):
         pi_code.append("    x = random() * 2 - 1\n")
         pi_code.append("    y = random() * 2 - 1\n")
         pi_code.append("    return 1 if x ** 2 + y ** 2 <= 1 else 0\n")
+        pi_code.append("pickled_f = cloudpickle.dumps(f)\n")
+        pi_code.append("print('Function f successfully pickled!')\n")
         # Use a lambda function directly in map to avoid complex serialization
         pi_code.append("count = sc.parallelize(range(1, n + 1), partitions).map(f).reduce(add)\n")
         pi_code.append('print("Pi is roughly %f" % (4.0 * count / n))\n')