Spaces:
Runtime error
Runtime error
File size: 1,572 Bytes
a5f8a35 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
#!/usr/bin/env python
import glob
import os
from setuptools import setup
import shutil
from typing import List
def get_model_zoo_configs() -> List[str]:
"""
Return a list of configs to include in package for model zoo. Copy over
these configs inside virtex/model_zoo.
"""
# Use absolute paths while symlinking.
source_configs_dir = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "configs"
)
destination = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "virtex", "model_zoo", "configs"
)
# Symlink the config directory inside package to have a cleaner pip install.
# Remove stale symlink/directory from a previous build.
if os.path.exists(source_configs_dir):
if os.path.islink(destination):
os.unlink(destination)
elif os.path.isdir(destination):
shutil.rmtree(destination)
if not os.path.exists(destination):
try:
os.symlink(source_configs_dir, destination)
except OSError:
# Fall back to copying if symlink fails: ex. on Windows.
shutil.copytree(source_configs_dir, destination)
config_paths = glob.glob("configs/**/*.yaml", recursive=True)
return config_paths
setup(
name="virtex",
version="1.1.0",
author="Karan Desai and Justin Johnson",
description="VirTex: Learning Visual Representations with Textual Annotations",
package_data={"virtex.model_zoo": get_model_zoo_configs()},
python_requires=">=3.6",
license="MIT",
zip_safe=True,
)
|