From bae631b9b48f8de54010522840669c4007c269c7 Mon Sep 17 00:00:00 2001
From: Rohith Krishna <rohith@localhost>
Date: Wed, 3 Dec 2025 02:35:36 -0800
Subject: [PATCH] force include yamls for inference

---
 models/rfd3/src/rfd3/cli.py | 14 ++++++++++----
 pyproject.toml              |  4 ++++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/models/rfd3/src/rfd3/cli.py b/models/rfd3/src/rfd3/cli.py
index c305687..fbeb7fa 100644
--- a/models/rfd3/src/rfd3/cli.py
+++ b/models/rfd3/src/rfd3/cli.py
@@ -12,10 +12,16 @@ app = typer.Typer()
 def design(ctx: typer.Context):
     """Run design using hydra config overrides and input files."""
     # Find the RFD3 configs directory relative to this file
-    # This file is at: models/rfd3/src/rfd3/cli.py
-    # Configs are at: models/rfd3/configs/
-    rfd3_package_dir = Path(__file__).parent.parent.parent  # Go up to models/rfd3/
-    config_path = str(rfd3_package_dir / "configs")
+    # Development: models/rfd3/src/rfd3/cli.py -> models/rfd3/configs/
+    # Installed: site-packages/rfd3/cli.py -> site-packages/rfd3/configs/
+
+    # Try development location first
+    dev_config_path = Path(__file__).parent.parent.parent / "configs"
+    if dev_config_path.exists():
+        config_path = str(dev_config_path)
+    else:
+        # Fall back to installed package location
+        config_path = str(Path(__file__).parent / "configs")
 
     # Get all arguments
     args = ctx.params.get("args", []) + ctx.args
diff --git a/pyproject.toml b/pyproject.toml
index 630d1d3..da32e28 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -115,6 +115,10 @@ packages = [
   "models/mpnn/src/mpnn",
 ]
 
+[tool.hatch.build.targets.wheel.force-include]
+"models/rfd3/configs" = "rfd3/configs"
+"models/rf3/configs" = "rf3/configs"
+
 # Formatting & linting settings -------------------------------------------------------
 [tool.ruff]
 line-length = 88