Skip to content

Commit

Permalink
update import for PyPi packing
Browse files Browse the repository at this point in the history
  • Loading branch information
themattinthehatt committed Apr 1, 2024
1 parent 3d0363a commit 5fddda4
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,11 @@

import re
import os
import subprocess

from setuptools import find_packages, setup

VERSION = "1.2.0"
VERSION = "1.2.1"

# add the README.md file to the long_description
with open("README.md", "r") as fh:
Expand Down Expand Up @@ -69,7 +70,7 @@ def get_cuda_version():
"typing",
dali,
# PyPI does not support direct dependencies, so we remove this line before uploading from PyPI
"segment_anything @ git+https://github.com/facebookresearch/segment-anything.git",
# "segment_anything @ git+https://github.com/facebookresearch/segment-anything.git",
]

# additional requirements
Expand All @@ -92,9 +93,9 @@ def get_cuda_version():
# collect all data and script files
data_files = []
for root, dirs, files in os.walk("lightning_pose/data"):
data_files.extend([join(root, f) for f in files])
data_files.extend([os.path.join(root, f) for f in files])
for root, dirs, files in os.walk("lightning_pose/scripts"):
data_files.extend([join(root, f) for f in files])
data_files.extend([os.path.join(root, f) for f in files])


setup(
Expand Down

0 comments on commit 5fddda4

Please sign in to comment.