Skip to content

Commit

Permalink
Fixup paths.
Browse files Browse the repository at this point in the history
  • Loading branch information
stellaraccident committed Feb 5, 2024
1 parent 81d292a commit bb97f31
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 12 deletions.
2 changes: 1 addition & 1 deletion core/mypy.ini
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
[mypy]

mypy_path = $MYPY_CONFIG_FILE_DIR/python
mypy_path = $MYPY_CONFIG_FILE_DIR
packages = shark_turbine.aot,shark_turbine.dynamo,shark_turbine.support
9 changes: 3 additions & 6 deletions core/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,12 +12,13 @@
from setuptools import find_namespace_packages, setup

THIS_DIR = os.path.realpath(os.path.dirname(__file__))
VERSION_INFO_FILE = os.path.join(THIS_DIR, "version_info.json")
REPO_DIR = os.path.dirname(THIS_DIR)
VERSION_INFO_FILE = os.path.join(REPO_DIR, "version_info.json")


with open(
os.path.join(
THIS_DIR,
REPO_DIR,
"README.md",
),
"rt",
Expand All @@ -38,7 +39,6 @@ def load_version_info():
"shark_turbine",
"shark_turbine.*",
],
where="python",
)

print("Found packages:", packages)
Expand Down Expand Up @@ -90,9 +90,6 @@ def initialize_options(self):
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
],
package_dir={
"": "python",
},
packages=packages,
entry_points={
"torch_dynamo_backends": [
Expand Down
6 changes: 1 addition & 5 deletions models/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@


TURBINE_MODELS_DIR = os.path.realpath(os.path.dirname(__file__))
TURBINE_ROOT_DIR = Path(TURBINE_MODELS_DIR).parent.parent
TURBINE_ROOT_DIR = Path(TURBINE_MODELS_DIR).parent
print(TURBINE_ROOT_DIR)
VERSION_INFO_FILE = os.path.join(TURBINE_ROOT_DIR, "version_info.json")

Expand Down Expand Up @@ -47,15 +47,11 @@ def load_version_info():
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
],
package_dir={
"": f"{str(TURBINE_ROOT_DIR)}/python",
},
packages=find_namespace_packages(
include=[
"turbine_models",
"turbine_models.*",
],
where=f"{str(TURBINE_ROOT_DIR)}/python",
),
install_requires=[
"Shark-Turbine",
Expand Down

0 comments on commit bb97f31

Please sign in to comment.