metaflow 2.14.3__py2.py3-none-any.whl → 2.15.1__py2.py3-none-any.whl

metaflow/plugins/pypi/parsers.py

@@ -0,0 +1,268 @@
+# this file can be overridden by extensions as is (e.g. metaflow-nflx-extensions)
+from metaflow.exception import MetaflowException
+
+
+class ParserValueError(MetaflowException):
+    headline = "Value error"
+
+
+def requirements_txt_parser(content: str):
+    """
+    Parse non-comment lines from a requirements.txt file as strictly valid
+    PEP 508 requirements.
+
+    Recognizes direct references (e.g. "my_lib @ git+https://..."), extras
+    (e.g. "requests[security]"), and version specifiers (e.g. "==2.0"). If
+    the package name is "python", its specifier is stored in the "python"
+    key instead of "packages".
+
+    Parameters
+    ----------
+    content : str
+        Contents of a requirements.txt file.
+
+    Returns
+    -------
+    dict
+        A dictionary with two keys:
+            - "packages": dict(str -> str)
+              Mapping from package name (plus optional extras/references) to a
+              version specifier string.
+            - "python": str or None
+              The Python version constraints if present, otherwise None.
+
+    Raises
+    ------
+     ParserValueError
+        If a requirement line is invalid PEP 508 or if environment markers are
+        detected, or if multiple Python constraints are specified.
+    """
+    import re
+    from metaflow._vendor.packaging.requirements import Requirement, InvalidRequirement
+
+    parsed = {"packages": {}, "python": None}
+
+    inline_comment_pattern = re.compile(r"\s+#.*$")
+    for line in content.splitlines():
+        line = line.strip()
+
+        # support Rye lockfiles by skipping lines not compliant with requirements
+        if line == "-e file:.":
+            continue
+
+        if not line or line.startswith("#"):
+            continue
+
+        line = inline_comment_pattern.sub("", line).strip()
+        if not line:
+            continue
+
+        try:
+            req = Requirement(line)
+        except InvalidRequirement:
+            raise ParserValueError(f"Not a valid PEP 508 requirement: '{line}'")
+
+        if req.marker is not None:
+            raise ParserValueError(
+                "Environment markers (e.g. 'platform_system==\"Linux\"') "
+                f"are not supported for line: '{line}'"
+            )
+
+        dep_key = req.name
+        if req.extras:
+            dep_key += f"[{','.join(req.extras)}]"
+        if req.url:
+            dep_key += f"@{req.url}"
+
+        dep_spec = str(req.specifier).lstrip(" =")
+
+        if req.name.lower() == "python":
+            if parsed["python"] is not None and dep_spec:
+                raise ParserValueError(
+                    f"Multiple Python version specs not allowed: '{line}'"
+                )
+            parsed["python"] = dep_spec or None
+        else:
+            parsed["packages"][dep_key] = dep_spec
+
+    return parsed
+
+
+def pyproject_toml_parser(content: str):
+    """
+    Parse a pyproject.toml file per PEP 621.
+
+    Reads the 'requires-python' and 'dependencies' fields from the "[project]" section.
+    Each dependency line must be a valid PEP 508 requirement. If the package name is
+    "python", its specifier is stored in the "python" key instead of "packages".
+
+    Parameters
+    ----------
+    content : str
+        Contents of a pyproject.toml file.
+
+    Returns
+    -------
+    dict
+        A dictionary with two keys:
+            - "packages": dict(str -> str)
+              Mapping from package name (plus optional extras/references) to a
+              version specifier string.
+            - "python": str or None
+              The Python version constraints if present, otherwise None.
+
+    Raises
+    ------
+    RuntimeError
+        If no TOML library (tomllib in Python 3.11+ or tomli in earlier versions) is found.
+     ParserValueError
+        If a dependency is not valid PEP 508, if environment markers are used, or if
+        multiple Python constraints are specified.
+    """
+    try:
+        import tomllib as toml  # Python 3.11+
+    except ImportError:
+        try:
+            import tomli as toml  # Python < 3.11 (requires "tomli" package)
+        except ImportError:
+            raise RuntimeError(
+                "Could not import a TOML library. For Python <3.11, please install 'tomli'."
+            )
+    from metaflow._vendor.packaging.requirements import Requirement, InvalidRequirement
+
+    data = toml.loads(content)
+
+    project = data.get("project", {})
+    requirements = project.get("dependencies", [])
+    requires_python = project.get("requires-python")
+
+    parsed = {"packages": {}, "python": None}
+
+    if requires_python is not None:
+        # If present, store verbatim; note that PEP 621 does not necessarily
+        # require "python" to be a dependency in the usual sense.
+        # Example: "requires-python" = ">=3.7,<4"
+        parsed["python"] = requires_python.lstrip("=").strip()
+
+    for dep_line in requirements:
+        dep_line_stripped = dep_line.strip()
+        try:
+            req = Requirement(dep_line_stripped)
+        except InvalidRequirement:
+            raise ParserValueError(
+                f"Not a valid PEP 508 requirement: '{dep_line_stripped}'"
+            )
+
+        if req.marker is not None:
+            raise ParserValueError(
+                f"Environment markers not supported for line: '{dep_line_stripped}'"
+            )
+
+        dep_key = req.name
+        if req.extras:
+            dep_key += f"[{','.join(req.extras)}]"
+        if req.url:
+            dep_key += f"@{req.url}"
+
+        dep_spec = str(req.specifier).lstrip("=")
+
+        if req.name.lower() == "python":
+            if parsed["python"] is not None and dep_spec:
+                raise ParserValueError(
+                    f"Multiple Python version specs not allowed: '{dep_line_stripped}'"
+                )
+            parsed["python"] = dep_spec or None
+        else:
+            parsed["packages"][dep_key] = dep_spec
+
+    return parsed
+
+
+def conda_environment_yml_parser(content: str):
+    """
+    Parse a minimal environment.yml file under strict assumptions.
+
+    The file must contain a 'dependencies:' line, after which each dependency line
+    appears with a '- ' prefix. Python can appear as 'python=3.9', etc.; other
+    packages as 'numpy=1.21.2' or simply 'numpy'. Non-compliant lines raise  ParserValueError.
+
+    Parameters
+    ----------
+    content : str
+        Contents of a environment.yml file.
+
+    Returns
+    -------
+    dict
+        A dictionary with keys:
+        {
+            "packages": dict(str -> str),
+            "python": str or None
+        }
+
+    Raises
+    ------
+     ParserValueError
+        If the file has malformed lines or unsupported sections.
+    """
+    import re
+
+    packages = {}
+    python_version = None
+
+    inside_dependencies = False
+
+    # Basic pattern for lines like "numpy=1.21.2"
+    # Group 1: package name
+    # Group 2: optional operator + version (could be "=1.21.2", "==1.21.2", etc.)
+    line_regex = re.compile(r"^([A-Za-z0-9_\-\.]+)(\s*[=<>!~].+\s*)?$")
+    inline_comment_pattern = re.compile(r"\s+#.*$")
+
+    for line in content.splitlines():
+        line = line.strip()
+        if not line or line.startswith("#"):
+            continue
+
+        line = inline_comment_pattern.sub("", line).strip()
+        if not line:
+            continue
+
+        if line.lower().startswith("dependencies:"):
+            inside_dependencies = True
+            continue
+
+        if inside_dependencies and not line.startswith("-"):
+            inside_dependencies = False
+            continue
+
+        if not inside_dependencies:
+            continue
+
+        dep_line = line.lstrip("-").strip()
+        if dep_line.endswith(":"):
+            raise ParserValueError(
+                f"Unsupported subsection '{dep_line}' in environment.yml."
+            )
+
+        match = line_regex.match(dep_line)
+        if not match:
+            raise ParserValueError(
+                f"Line '{dep_line}' is not a valid conda package specifier."
+            )
+
+        pkg_name, pkg_version_part = match.groups()
+        version_spec = pkg_version_part.strip() if pkg_version_part else ""
+
+        if version_spec.startswith("="):
+            version_spec = version_spec.lstrip("=").strip()
+
+        if pkg_name.lower() == "python":
+            if python_version is not None and version_spec:
+                raise ParserValueError(
+                    f"Multiple Python version specs detected: '{dep_line}'"
+                )
+            python_version = version_spec
+        else:
+            packages[pkg_name] = version_spec
+
+    return {"packages": packages, "python": python_version}

metaflow/plugins/pypi/utils.py

@@ -72,6 +72,24 @@ def pip_tags(python_version, mamba_platform):
             )
         ]
         platforms.append("linux_x86_64")
+    elif mamba_platform == "linux-aarch64":
+        platforms = [
+            "manylinux%s_aarch64" % s
+            for s in (
+                "2014",
+                "_2_17",
+                "_2_18",
+                "_2_19",
+                "_2_20",
+                "_2_21",
+                "_2_23",
+                "_2_24",
+                "_2_25",
+                "_2_26",
+                "_2_27",
+            )
+        ]
+        platforms.append("linux_aarch64")
     elif mamba_platform == "osx-64":
         platforms = tags.mac_platforms(arch="x86_64")
     elif mamba_platform == "osx-arm64":

metaflow/runner/subprocess_manager.py

@@ -78,13 +78,19 @@ class SubprocessManager(object):
         self.commands: Dict[int, CommandManager] = {}
 
         try:
-            loop = asyncio.get_running_loop()
-            loop.add_signal_handler(
-                signal.SIGINT,
-                lambda: asyncio.create_task(self._async_handle_sigint()),
+            try:
+                loop = asyncio.get_running_loop()
+                loop.add_signal_handler(
+                    signal.SIGINT,
+                    lambda: asyncio.create_task(self._async_handle_sigint()),
+                )
+            except RuntimeError:
+                signal.signal(signal.SIGINT, self._handle_sigint)
+        except ValueError:
+            sys.stderr.write(
+                "Warning: Unable to set signal handlers in non-main thread. "
+                "Interrupt handling will be limited.\n"
             )
-        except RuntimeError:
-            signal.signal(signal.SIGINT, self._handle_sigint)
 
     async def _async_handle_sigint(self):
         pids = [

metaflow/version.py

@@ -1 +1 @@
-metaflow_version = "2.14.3"
+metaflow_version = "2.15.1"

metaflow-2.15.1.data/data/share/metaflow/devtools/Makefile

@@ -0,0 +1,322 @@
+SHELL := /bin/bash
+.SHELLFLAGS := -eu -o pipefail -c
+
+help:
+	@echo "Available targets:"
+	@echo "  up            - Start the development environment"
+	@echo "  shell         - Switch to development environment's shell"
+	@echo "  ui            - Open Metaflow UI"
+	@echo "  dashboard     - Open Minikube dashboard"
+	@echo "  down          - Stop and clean up the environment"
+	@echo "  help          - Show this help message"
+
+HELM_VERSION := v3.14.0
+MINIKUBE_VERSION := v1.32.0
+TILT_VERSION := v0.33.11
+GUM_VERSION := v0.15.2
+
+MKFILE_PATH := $(abspath $(lastword $(MAKEFILE_LIST)))
+MKFILE_DIR := $(dir $(MKFILE_PATH))
+DEVTOOLS_DIR := $(MKFILE_DIR).devtools
+PICK_SERVICES := $(MKFILE_DIR)pick_services.sh
+MINIKUBE_DIR := $(DEVTOOLS_DIR)/minikube
+MINIKUBE := $(MINIKUBE_DIR)/minikube
+TILT_DIR := $(DEVTOOLS_DIR)/tilt
+TILT := $(TILT_DIR)/tilt
+TILTFILE := $(MKFILE_DIR)/Tiltfile
+MAKE_CMD := $(MAKE) -C "$(MKFILE_DIR)"
+
+MINIKUBE_CPUS ?= 4
+MINIKUBE_MEMORY ?= 6000
+MINIKUBE_DISK_SIZE ?= 20g
+
+ifeq ($(shell uname), Darwin)
+	minikube_os = darwin
+	tilt_os = mac
+else
+	minikube_os = linux
+	tilt_os = linux
+endif
+
+ifeq ($(shell uname -m), x86_64)
+	arch = amd64
+	tilt_arch = x86_64
+else
+	arch = arm64
+	tilt_arch = arm64
+endif
+
+# TODO: Move scripts to a folder
+
+install-helm:
+	@if ! command -v helm >/dev/null 2>&1; then \
+		echo "📥 Installing Helm $(HELM_VERSION) (may require sudo access)..."; \
+		curl -fsSL https://raw.githubusercontent.com/helm/helm/main/scripts/get-helm-3 | HELM_INSTALL_VERSION=$(HELM_VERSION) bash; \
+		echo "✅ Helm installation complete"; \
+	fi
+
+check-docker:
+	@if ! command -v docker >/dev/null 2>&1; then \
+		echo "❌ Docker is not installed. Please install Docker first: https://docs.docker.com/get-docker/"; \
+		exit 1; \
+	fi
+	@echo "🔍 Checking Docker daemon..."
+	@if [ "$(shell uname)" = "Darwin" ]; then \
+		open -a Docker || (echo "❌ Please start Docker Desktop" && exit 1); \
+	else \
+		systemctl is-active --quiet docker || (echo "❌ Docker daemon is not running. Start with 'sudo systemctl start docker'" && exit 1); \
+	fi
+	@echo "✅ Docker is running"
+
+install-brew:
+	@if [ "$(shell uname)" = "Darwin" ] && ! command -v brew >/dev/null 2>&1; then \
+		echo "📥 Installing Homebrew..."; \
+		/bin/bash -c "$$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"; \
+		echo "✅ Homebrew installation complete"; \
+	fi
+
+install-curl:
+	@if ! command -v curl >/dev/null 2>&1; then \
+		echo "📥 Installing curl..."; \
+		if [ "$(shell uname)" = "Darwin" ]; then \
+			HOMEBREW_NO_AUTO_UPDATE=1 brew install curl; \
+		elif command -v apt-get >/dev/null 2>&1; then \
+			sudo apt-get update && sudo apt-get install -y curl; \
+		elif command -v yum >/dev/null 2>&1; then \
+			sudo yum install -y curl; \
+		elif command -v dnf >/dev/null 2>&1; then \
+			sudo dnf install -y curl; \
+		else \
+			echo "❌ Could not install curl. Please install manually."; \
+			exit 1; \
+		fi; \
+		echo "✅ curl installation complete"; \
+	fi
+
+install-gum:
+	@echo "🔍 Checking if gum is installed..."
+	@if ! command -v gum >/dev/null 2>&1; then \
+		echo "📥 Installing gum..."; \
+		if [ "$(shell uname)" = "Darwin" ]; then \
+			HOMEBREW_NO_AUTO_UPDATE=1 brew install gum; \
+		elif command -v apt-get >/dev/null 2>&1; then \
+			curl -fsSL -o /tmp/gum.deb \
+			"https://github.com/charmbracelet/gum/releases/download/$(GUM_VERSION)/gum_$(GUM_VERSION:v%=%)_$(arch).deb"; \
+			sudo apt-get update -qq; \
+			sudo apt-get install -y /tmp/gum.deb || sudo dpkg -i /tmp/gum.deb; \
+			rm -f /tmp/gum.deb; \
+		else \
+			echo "❌ Could not determine how to install gum for your platform. Please install manually."; \
+			exit 1; \
+		fi; \
+		echo "✅ gum installation complete"; \
+	else \
+		echo "✅ gum is already installed."; \
+	fi
+
+setup-minikube:
+	@if [ ! -f "$(MINIKUBE)" ]; then \
+		echo "📥 Installing Minikube $(MINIKUBE_VERSION)"; \
+		mkdir -p $(MINIKUBE_DIR); \
+		curl -L --fail https://github.com/kubernetes/minikube/releases/download/$(MINIKUBE_VERSION)/minikube-$(minikube_os)-$(arch) -o $(MINIKUBE) || (echo "❌ Failed to download minikube" && exit 1); \
+		chmod +x $(MINIKUBE); \
+		echo "✅ Minikube $(MINIKUBE_VERSION) installed successfully"; \
+	fi
+	@echo "🔧 Setting up Minikube $(MINIKUBE_VERSION) cluster..."
+	@if ! $(MINIKUBE) status >/dev/null 2>&1; then \
+		echo "🚀 Starting new Minikube $(MINIKUBE_VERSION) cluster..."; \
+		$(MINIKUBE) start \
+			--cpus $(MINIKUBE_CPUS) \
+			--memory $(MINIKUBE_MEMORY) \
+			--disk-size $(MINIKUBE_DISK_SIZE) \
+			--driver docker; \
+		echo "🔌 Enabling metrics-server and dashboard (quietly)..."; \
+		$(MINIKUBE) addons enable metrics-server >/dev/null 2>&1; \
+		$(MINIKUBE) addons enable dashboard >/dev/null 2>&1; \
+	else \
+		echo "✅ Minikube $(MINIKUBE_VERSION) cluster is already running"; \
+	fi
+	@echo "🎉 Minikube $(MINIKUBE_VERSION) cluster is ready!"
+
+setup-tilt:
+	@if [ ! -f "$(TILT)" ]; then \
+		echo "📥 Installing Tilt $(TILT_VERSION)"; \
+		mkdir -p $(TILT_DIR); \
+		(curl -L https://github.com/tilt-dev/tilt/releases/download/$(TILT_VERSION)/tilt.$(TILT_VERSION:v%=%).$(tilt_os).$(tilt_arch).tar.gz | tar -xz -C $(TILT_DIR)) && echo "✅ Tilt $(TILT_VERSION) installed successfully" || (echo "❌ Failed to install Tilt" && exit 1); \
+	fi
+
+tunnel:
+	$(MINIKUBE) tunnel
+
+teardown-minikube:
+	@echo "🛑 Stopping Minikube $(MINIKUBE_VERSION) cluster..."
+	-$(MINIKUBE) stop
+	@echo "🗑️  Deleting Minikube $(MINIKUBE_VERSION) cluster..."
+	-$(MINIKUBE) delete --all
+	@echo "🧹 Removing Minikube binary..."
+	-rm -rf $(MINIKUBE_DIR)
+	@echo "✅ Minikube $(MINIKUBE_VERSION) teardown complete"
+
+dashboard:
+	@echo "🔗 Opening Minikube Dashboard..."
+	@$(MINIKUBE) dashboard
+
+# make shell is symlinked to metaflow-dev shell by metaflow
+up: install-brew check-docker install-curl install-gum setup-minikube install-helm setup-tilt
+	@echo "🚀 Starting up (may require sudo access)..."
+	@mkdir -p $(DEVTOOLS_DIR)
+	@echo '#!/bin/bash' > $(DEVTOOLS_DIR)/start.sh
+	@echo 'set -e' >> $(DEVTOOLS_DIR)/start.sh
+	@echo 'trap "exit" INT TERM' >> $(DEVTOOLS_DIR)/start.sh
+	@echo 'trap "kill 0" EXIT' >> $(DEVTOOLS_DIR)/start.sh
+	@echo 'eval $$($(MINIKUBE) docker-env)' >> $(DEVTOOLS_DIR)/start.sh
+	
+	@echo 'if [ -n "$$SERVICES_OVERRIDE" ]; then' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo '    echo "🌐 Using user-provided list of services: $$SERVICES_OVERRIDE"' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo '    SERVICES="$$SERVICES_OVERRIDE"' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo 'else' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo '    echo "📝 Selecting services..."' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo '    SERVICES=$$($(PICK_SERVICES))' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo 'fi' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo 'PATH="$(MINIKUBE_DIR):$(TILT_DIR):$$PATH" $(MINIKUBE) tunnel &' >> $(DEVTOOLS_DIR)/start.sh
+	@echo 'echo -e "🚀 Starting Tilt with selected services..."' >> $(DEVTOOLS_DIR)/start.sh
+	@echo 'echo -e "\033[1;38;5;46m\n🔥 \033[1;38;5;196mNext Steps:\033[0;38;5;46m Use \033[3mmetaflow-dev shell\033[23m to switch to the development\n   environment'\''s shell and start executing your Metaflow flows.\n\033[0m"' >> "$(DEVTOOLS_DIR)/start.sh"
+	@echo 'PATH="$(MINIKUBE_DIR):$(TILT_DIR):$$PATH" SERVICES="$$SERVICES" tilt up -f $(TILTFILE)' >> $(DEVTOOLS_DIR)/start.sh
+	@echo 'wait' >> $(DEVTOOLS_DIR)/start.sh
+	@chmod +x $(DEVTOOLS_DIR)/start.sh
+	@$(DEVTOOLS_DIR)/start.sh
+
+down:
+	@echo "🛑 Stopping all services..."
+	@-pkill -f "$(MINIKUBE) tunnel" 2>/dev/null || true
+	@echo "⏹️  Stopping Tilt..."
+	-PATH="$(MINIKUBE_DIR):$(TILT_DIR):$$PATH" tilt down -f $(TILTFILE)
+	@echo "🧹 Cleaning up Minikube..."
+	$(MAKE_CMD) teardown-minikube
+	@echo "🗑️  Removing Tilt binary and directory..."
+	-rm -rf $(TILT_DIR)
+	@echo "🧹 Removing temporary scripts..."
+	-rm -rf $(DEVTOOLS_DIR)
+	@echo "✨ All done!"
+
+shell: setup-tilt
+	@echo "⏳ Checking if development environment is up..."
+	@set -e; \
+	for i in $$(seq 1 90); do \
+		if "$(TILT)" get session >/dev/null 2>&1; then \
+			found_session=1; \
+			break; \
+		else \
+			sleep 2; \
+		fi; \
+	done; \
+	if [ -z "$${found_session}" ]; then \
+		echo "❌ Development environment is not up."; \
+		echo "   Please run 'metaflow-dev up' in another terminal, then re-run 'metaflow-dev shell'."; \
+		exit 1; \
+	fi
+	@echo "⏳ Waiting for development environment to be ready..."
+	@while true; do \
+		"$(TILT)" get uiresource generate-configs >/dev/null 2>&1; \
+		status=$$?; \
+		if [ $$status -eq 0 ]; then \
+			"$(TILT)" wait --for=condition=Ready uiresource/generate-configs; \
+			break; \
+		elif [ $$status -eq 127 ]; then \
+			echo "❌ Development environment is not up."; \
+			echo "   Please run 'metaflow-dev up' in another terminal, then re-run 'metaflow-dev shell'."; \
+			exit 1; \
+		else \
+			sleep 1; \
+		fi; \
+	done
+	@echo "🔧 Starting a new shell for development environment..."
+	@bash -c '\
+		if [ -n "$$SHELL" ]; then \
+			user_shell="$$SHELL"; \
+		else \
+			user_shell="$(SHELL)"; \
+		fi; \
+		echo "🔎 Using $$user_shell for interactive session."; \
+		echo "🐍 If you installed Metaflow in a virtual environment, activate it now."; \
+		if [ -f "$(DEVTOOLS_DIR)/aws_config" ]; then \
+			env METAFLOW_HOME="$(DEVTOOLS_DIR)" \
+				METAFLOW_PROFILE=local \
+				AWS_CONFIG_FILE="$(DEVTOOLS_DIR)/aws_config" \
+				"$$user_shell" -i; \
+		else \
+			env METAFLOW_HOME="$(DEVTOOLS_DIR)" \
+				METAFLOW_PROFILE=local \
+				"$$user_shell" -i; \
+		fi'
+
+
+# @echo '$(MAKE_CMD) create-dev-shell' >> $(DEVTOOLS_DIR)/start.sh
+# @echo 'rm -f /tmp/metaflow-devshell-*' >> $(DEVTOOLS_DIR)/start.sh
+create-dev-shell: setup-tilt
+	@bash -c '\
+		SHELL_PATH=/tmp/metaflow-dev-shell-$$$$ && \
+		echo "#!/bin/bash" > $$SHELL_PATH && \
+		echo "set -e" >> $$SHELL_PATH && \
+		echo "" >> $$SHELL_PATH && \
+		echo "echo \"⏳ Checking if development environment is up...\"" >> $$SHELL_PATH && \
+		echo "if ! $(TILT) get session >/dev/null 2>&1; then" >> $$SHELL_PATH && \
+		echo "  echo \"❌ Development environment is not up.\"" >> $$SHELL_PATH && \
+		echo "  echo \"   Please run '\''make up'\'' in another terminal, then re-run this script.\"" >> $$SHELL_PATH && \
+		echo "  exit 1" >> $$SHELL_PATH && \
+		echo "fi" >> $$SHELL_PATH && \
+		echo "" >> $$SHELL_PATH && \
+		echo "echo \"⏳ Waiting for development environment to be ready...\"" >> $$SHELL_PATH && \
+		echo "$(TILT) wait --for=condition=Ready uiresource/generate-configs" >> $$SHELL_PATH && \
+		echo "" >> $$SHELL_PATH && \
+		echo "echo \"🔧 Starting a new shell for development environment...\"" >> $$SHELL_PATH && \
+		echo "if [ -n \"\$$SHELL\" ]; then" >> $$SHELL_PATH && \
+		echo "    user_shell=\"\$$SHELL\"" >> $$SHELL_PATH && \
+		echo "else" >> $$SHELL_PATH && \
+		echo "    user_shell=\"$(SHELL)\"" >> $$SHELL_PATH && \
+		echo "fi" >> $$SHELL_PATH && \
+		echo "echo \"🔎 Using \$$user_shell for interactive session.\"" >> $$SHELL_PATH && \
+		echo "echo \"🐍 If you installed Metaflow in a virtual environment, activate it now.\"" >> $$SHELL_PATH && \
+		echo "if [ -f \"$(DEVTOOLS_DIR)/aws_config\" ]; then" >> $$SHELL_PATH && \
+		echo "    env METAFLOW_HOME=\"$(DEVTOOLS_DIR)\" \\" >> $$SHELL_PATH && \
+		echo "        METAFLOW_PROFILE=local \\" >> $$SHELL_PATH && \
+		echo "        AWS_CONFIG_FILE=\"$(DEVTOOLS_DIR)/aws_config\" \\" >> $$SHELL_PATH && \
+		echo "        \"\$$user_shell\" -i" >> $$SHELL_PATH && \
+		echo "else" >> $$SHELL_PATH && \
+		echo "    env METAFLOW_HOME=\"$(DEVTOOLS_DIR)\" \\" >> $$SHELL_PATH && \
+		echo "        METAFLOW_PROFILE=local \\" >> $$SHELL_PATH && \
+		echo "        \"\$$user_shell\" -i" >> $$SHELL_PATH && \
+		echo "fi" >> $$SHELL_PATH && \
+		chmod +x $$SHELL_PATH && \
+		echo "✨ Created $$SHELL_PATH" && \
+		echo "🔑  Execute it from ANY directory to switch to development environment shell!" \
+	'
+
+ui: setup-tilt
+	@echo "⏳ Checking if the development environment is up..."
+	@if ! $(TILT) get session >/dev/null 2>&1; then \
+		echo "❌ Development environment is not up."; \
+		echo "   Please run 'metaflow-dev up' in another terminal, then re-run 'metaflow-dev ui'."; \
+		exit 1; \
+	fi
+	@echo "⏳ Waiting for Metaflow UI to be ready..."
+	@while true; do \
+		"$(TILT)" get uiresource metaflow-ui >/dev/null 2>&1; \
+		status=$$?; \
+		if [ $$status -eq 0 ]; then \
+			"$(TILT)" wait --for=condition=Ready uiresource/metaflow-ui; \
+			break; \
+		elif [ $$status -eq 127 ]; then \
+			echo "❌ Development environment is not up."; \
+			echo "   Please run 'metaflow-dev up' in another terminal, then re-run 'metaflow-dev shell'."; \
+			exit 1; \
+		else \
+			sleep 1; \
+		fi; \
+	done
+	@echo "🔗 Opening Metaflow UI at http://localhost:3000"
+	@open http://localhost:3000
+
+.PHONY: install-helm setup-minikube setup-tilt teardown-minikube tunnel up down check-docker install-curl install-gum install-brew up down dashboard shell ui help
+
+.DEFAULT_GOAL := up