force include yamls for inference

This commit is contained in:
Rohith Krishna
2025-12-03 02:35:36 -08:00
parent 86137ef3ed
commit bae631b9b4
2 changed files with 14 additions and 4 deletions

View File

@@ -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

View File

@@ -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