fix: run pyenv update

metaopt · Oct 30, 2024 · 96b6595 · 96b6595
1 parent edd6cbe
commit 96b6595
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.github/workflows/tests-with-pydebug.yml b/.github/workflows/tests-with-pydebug.yml
@@ -91,6 +91,9 @@ jobs:
       - name: Determine Python version
         shell: bash
         run: |
+          if [[ "${{ runner.os }}" == 'Windows' ]]; then
+            pyenv update
+          fi
           pyenv install --list
           if [[ "${{ matrix.python-abiflags }}" == *t* ]]; then
             PYTHON_VERSION="$(
@@ -122,6 +125,7 @@ jobs:
               PYENV_INSTALL_ARGS+=("--debug")
             fi
           else
+            pyenv update
             if [[ "${{ matrix.python-abiflags }}" == *d* ]]; then
               PYENV_INSTALL_ARGS+=("--dev")
             fi

diff --git a/optree/registry.py b/optree/registry.py
@@ -147,8 +147,9 @@ def _pytree_node_registry_get(  # pylint: disable=too-many-return-statements
     *,
     namespace: str = '',
 ) -> PyTreeNodeRegistryEntry | None:
-    handler: PyTreeNodeRegistryEntry | None = None
-    if namespace is not __GLOBAL_NAMESPACE and namespace != '':
+    if namespace is __GLOBAL_NAMESPACE:
+        namespace = ''
+    elif namespace != '':
         handler = _NODETYPE_REGISTRY.get((namespace, cls))
         if handler is not None:
             return handler