Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Create Transforms 1.0.0a0 first alpha release with new refactored transforms #874

Closed
wants to merge 156 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
156 commits
Select commit Hold shift + click to select a range
5ca102f
html2parquet example
touma-I Nov 13, 2024
ee1b3c5
fix typo with package name
touma-I Nov 13, 2024
244a7b3
remove output folder from git
touma-I Nov 16, 2024
3b8576d
Merge branch 'dev' into html2parquet-simplify
touma-I Nov 16, 2024
21da76b
Merge branch 'dev' into html2parquet-example
touma-I Nov 16, 2024
402cfb9
first draft make test-src working
touma-I Nov 17, 2024
078179a
fix docker build for all 3 runtime
touma-I Nov 17, 2024
a222cb3
fix publish-image
touma-I Nov 17, 2024
d3df908
use module name to invoke entry point
touma-I Nov 17, 2024
00c2d70
added kind-load-image required for workflow test
touma-I Nov 17, 2024
c128893
fix setting for transform runtime being used. The global setting is n…
touma-I Nov 17, 2024
b44fff1
override DOCKER_REMOTE_IMAGE defined in .make.defaults
touma-I Nov 17, 2024
6631851
chasing after bug with DOCKER_REMOTE_IMAGE value
touma-I Nov 17, 2024
1c58a5c
chasing after DOCKER_REMOTE_IMAGE
touma-I Nov 17, 2024
981930c
bug fixes
touma-I Nov 18, 2024
3481667
odify notebook with simplified API
touma-I Nov 18, 2024
890a9da
keep for comparaison
touma-I Nov 18, 2024
e99e0bc
fix typo
touma-I Nov 18, 2024
590f5dd
added __init__
touma-I Nov 18, 2024
777fb9d
no longer needed
touma-I Nov 18, 2024
d7dfa3e
Addd test-image-python/ray/spark
touma-I Nov 19, 2024
f7fc598
first cut at refactoring pdf2parquet
touma-I Nov 19, 2024
a2835b3
added notebook and bug fixes
touma-I Nov 20, 2024
6cda6fb
fix typo
Nov 24, 2024
3c0187c
merge with dev with latest update for readme.md and notebook
touma-I Nov 24, 2024
8c8f6e8
first cut at refactoring doc_chunk
touma-I Nov 24, 2024
63051cb
-ssome cleanup
touma-I Nov 24, 2024
012e254
first cut refactoring dpk_text_encoder
touma-I Nov 24, 2024
7202347
fix type
touma-I Nov 25, 2024
3d7cb02
added runtime notebooks
touma-I Nov 25, 2024
38f7e25
merge with latest from dev
touma-I Nov 27, 2024
7b2381d
merge with dev
touma-I Dec 4, 2024
9a28f50
update notebook for new structure
touma-I Dec 4, 2024
07345e6
merge with dev
touma-I Dec 4, 2024
5052622
refactored doc quality transform as its own module
touma-I Dec 4, 2024
70cc5a7
use latest make targets
touma-I Dec 4, 2024
ae79ab5
Enhance make target
touma-I Dec 4, 2024
d3bd5ec
merge with dev
touma-I Dec 4, 2024
3c9b999
refactored code as its own module
touma-I Dec 5, 2024
49a22ae
added __init__
touma-I Dec 5, 2024
24b6d9d
merge with latest from dev
touma-I Dec 5, 2024
808521a
Fix typo
touma-I Dec 5, 2024
b8f3b6f
remove spark unit test for now
touma-I Dec 5, 2024
3baad3e
Show example for runnign ray runtime
touma-I Dec 5, 2024
37881ef
fixing issues with spark
touma-I Dec 5, 2024
9c869ee
remove BASE_IMAGE arg from dockerfile.spark
touma-I Dec 6, 2024
6fa0c04
added login to quay.io
touma-I Dec 6, 2024
bbe9a02
debug registry credential
touma-I Dec 6, 2024
b77aaa3
use dpk secrets
touma-I Dec 6, 2024
2c52fd5
testing registry user
touma-I Dec 6, 2024
5ffadb4
testing registry user
touma-I Dec 6, 2024
8b234a4
testing environment secrets
touma-I Dec 6, 2024
878fa42
testing environment secrets
touma-I Dec 6, 2024
5632c06
testing environment secrets
touma-I Dec 6, 2024
6da893f
testing environment secrets
touma-I Dec 6, 2024
d426439
testing environment secrets
touma-I Dec 6, 2024
e8bb04a
Delete .github/workflows/test-universal-doc_id.yml
touma-I Dec 6, 2024
3647ebc
restore workflow file
touma-I Dec 6, 2024
adaf78f
clear testing of docker login
touma-I Dec 6, 2024
fda7999
first cut at refactoring with own dpk_lang_id name space
touma-I Dec 9, 2024
b50988b
restore missing file
touma-I Dec 9, 2024
7836832
-sAdded README.md by combing python and ray
touma-I Dec 9, 2024
22d844a
README changes
shahrokhDaijavad Dec 9, 2024
af0f89b
Refactor hap transform as its own module
touma-I Dec 10, 2024
59a4fba
update cicd make file
touma-I Dec 10, 2024
0c38a4c
fix Makefile failing targets
touma-I Dec 10, 2024
dd3a63e
Merge branch 'dev' into hap-simplify
touma-I Dec 10, 2024
64781b9
fix notebook
touma-I Dec 10, 2024
fd0b261
merged with dev
touma-I Dec 10, 2024
e4ad78e
README changes
shahrokhDaijavad Dec 10, 2024
55dcce1
README changes
shahrokhDaijavad Dec 10, 2024
07b91ec
README changes
shahrokhDaijavad Dec 10, 2024
87eecf0
README changes
shahrokhDaijavad Dec 10, 2024
3b2420b
fix notebook
touma-I Dec 10, 2024
30912d1
More changes to README
shahrokhDaijavad Dec 10, 2024
2e324d4
README changes
shahrokhDaijavad Dec 10, 2024
62166d9
More README changes
shahrokhDaijavad Dec 10, 2024
02bc21a
added python notebook
touma-I Dec 10, 2024
ca3efdb
README changes
shahrokhDaijavad Dec 10, 2024
8ac2d8c
fix Makefile cli target although not very useful
touma-I Dec 10, 2024
d1ad598
README changes
shahrokhDaijavad Dec 10, 2024
0a0e785
added notebook and fix makefile
touma-I Dec 10, 2024
411bc19
test notebook with new simplified API
touma-I Dec 11, 2024
ac9b954
added hap extra to pip install
touma-I Dec 11, 2024
83d7e7c
fix pip install for notebook
touma-I Dec 11, 2024
97bc8f2
fix notebooks
touma-I Dec 11, 2024
127b290
fix notebook
touma-I Dec 11, 2024
0cbe63f
fix notebook
touma-I Dec 11, 2024
6c54442
added notebooks
touma-I Dec 11, 2024
0722d63
first cut at refactoring tokenization
touma-I Dec 11, 2024
18d0c46
fix script string
touma-I Dec 11, 2024
cbb68bc
added notebook
touma-I Dec 11, 2024
6f0dfef
README changes
shahrokhDaijavad Dec 11, 2024
73a69be
README fix of the link to the notebook
shahrokhDaijavad Dec 11, 2024
c59e5bb
fix typo in script name
touma-I Dec 11, 2024
43d792a
Merge branch 'dev' into pdf2parquet-simplify
touma-I Dec 11, 2024
c747870
merge text_encoder
touma-I Dec 11, 2024
36a00d2
added doc_chunk
touma-I Dec 11, 2024
419e80c
Merge branch 'docid-simplify' into transforms-1.0
touma-I Dec 11, 2024
29fc854
Merge branch 'docquality-simplify' into transforms-1.0
touma-I Dec 11, 2024
f66f4d4
draft pyproject
touma-I Dec 11, 2024
10fab76
Merge branch 'hap-simplify' into transforms-1.0
touma-I Dec 11, 2024
2187e97
Merge branch 'html2parquet-simplify' into transforms-1.0
touma-I Dec 11, 2024
1b03de7
Merge branch 'langid-simplify' into transforms-1.0
touma-I Dec 11, 2024
97c5919
Merge branch 'tokenization-simplify' into transforms-1.0
touma-I Dec 11, 2024
a35d8a3
build alpha release with all refactored transforms
touma-I Dec 12, 2024
ecabbd6
first cut a refactoring ededup as its own namespace dpk_ededup
touma-I Dec 12, 2024
4e28866
-supdate with latest cicd target
touma-I Dec 12, 2024
bb92881
added notebooks
touma-I Dec 12, 2024
44d44cb
Merge branch 'ededup-simplify' into transforms-1.0
touma-I Dec 12, 2024
2d4e3b4
fix notebook:
touma-I Dec 12, 2024
0cba14f
Merge branch 'doc_chunk-simplify' into transforms-1.0
touma-I Dec 12, 2024
cabd577
fix notebook
touma-I Dec 12, 2024
370a257
Merge branch 'docid-simplify' into transforms-1.0
touma-I Dec 12, 2024
a26715b
fix notebook
touma-I Dec 12, 2024
2ecaac2
Merge branch 'docquality-simplify' into transforms-1.0
touma-I Dec 12, 2024
f877db4
fix notebook
touma-I Dec 12, 2024
c9cc02c
Merge branch 'hap-simplify' into transforms-1.0
touma-I Dec 12, 2024
f5940d1
fix notebook
touma-I Dec 12, 2024
baf1e53
Merge branch 'html2parquet-simplify' into transforms-1.0
touma-I Dec 12, 2024
42d3587
fix notebook
touma-I Dec 12, 2024
fae3415
Merge branch 'langid-simplify' into transforms-1.0
touma-I Dec 12, 2024
2aff557
fix notebook
touma-I Dec 12, 2024
100f875
Merge branch 'pdf2parquet-simplify' into transforms-1.0
touma-I Dec 12, 2024
70e2506
fix notebook
touma-I Dec 12, 2024
32be8a0
Merge branch 'text_encoder-simplify' into transforms-1.0
touma-I Dec 12, 2024
c251033
fix notebook
touma-I Dec 12, 2024
63714ef
Merge branch 'tokenization-simplify' into transforms-1.0
touma-I Dec 12, 2024
6a4a4f5
added ededup
touma-I Dec 12, 2024
4750dd7
added ray notebook
touma-I Dec 12, 2024
ea4d5e1
Merge branch 'docid-simplify' into transforms-1.0
touma-I Dec 12, 2024
fbb6344
new alpha tag
touma-I Dec 12, 2024
85fb99a
README fixes
shahrokhDaijavad Dec 12, 2024
59f1ea8
One more README fix
shahrokhDaijavad Dec 12, 2024
2eb47bd
Added the link to the Ray notebook in README
shahrokhDaijavad Dec 12, 2024
74deeef
new iteration on pypi build
touma-I Dec 13, 2024
cfa0374
fix cli-sample target
touma-I Dec 13, 2024
0bdf75c
Merge branch 'ededup-simplify' into transforms-1.0
touma-I Dec 13, 2024
5d7f901
fix sample target
matouma Dec 13, 2024
b2ce1b6
fix sample target
matouma Dec 13, 2024
aa97a17
fix sample targets
matouma Dec 13, 2024
7ee946e
Merge remote-tracking branch 'origin/docid-simplify' into transforms-1.0
matouma Dec 13, 2024
0e2cce4
Merge remote-tracking branch 'origin/docquality-simplify' into transf…
matouma Dec 13, 2024
841af15
Merge remote-tracking branch 'origin/html2parquet-example' into trans…
matouma Dec 13, 2024
e07e76e
Merge remote-tracking branch 'origin/langid-simplify' into transforms…
matouma Dec 13, 2024
44f72ff
added dpk_ededup
matouma Dec 13, 2024
07bc76b
fix script string
matouma Dec 13, 2024
cf4d8a6
Merge branch 'doc_chunk-simplify' into transforms-1.0
matouma Dec 13, 2024
97219cd
steps notebooks
matouma Dec 13, 2024
ddd2083
added __init__.py
matouma Dec 14, 2024
eb5db4e
Merge branch 'ededup-simplify' into transforms-1.0
matouma Dec 14, 2024
4ba2605
added language only extra covering refactored transforms
matouma Dec 14, 2024
9c9e76b
restructure notebooks for 1.0 alpha release as a single one
matouma Dec 14, 2024
7965b5e
fix wf exec script
matouma Dec 14, 2024
e144927
fix pdf2parquet script
matouma Dec 14, 2024
88ff060
Merge branch 'pdf2parquet-simplify' into transforms-1.0
matouma Dec 14, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
89 changes: 52 additions & 37 deletions transforms/.make.cicd.targets
Original file line number Diff line number Diff line change
Expand Up @@ -51,63 +51,78 @@ publish:

test-image-sequence:: .defaults.lib-whl-image .transforms.test-image-help .transforms.clean

test-image-python:
$(MAKE) BUILD_SPECIFIC_RUNTIME=python test-image

test-image-ray:
$(MAKE) BUILD_SPECIFIC_RUNTIME=ray test-image

test-image-spark:
$(MAKE) BUILD_SPECIFIC_RUNTIME=spark test-image

test-image:: .default.build-lib-wheel
@if [ -e Dockerfile.python ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.python \
TRANSFORM_RUNTIME_SRC_FILE=$(TRANSFORM_PYTHON_SRC) \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-python \
test-image-sequence ; \
@if [ -z "$(BUILD_SPECIFIC_RUNTIME)" ] || [ "$(BUILD_SPECIFIC_RUNTIME)" == "python" ]; then \
if [ -e Dockerfile.python ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.python \
TRANSFORM_RUNTIME_SRC_FILE=$(TRANSFORM_PYTHON_SRC) \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-python \
test-image-sequence ; \
fi ;\
fi
@if [ -e Dockerfile.ray ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.ray \
TRANSFORM_RUNTIME_SRC_FILE=$(TRANSFORM_RAY_SRC) \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-ray \
BASE_IMAGE=$(RAY_BASE_IMAGE) \
test-image-sequence ; \
@if [ -z "$(BUILD_SPECIFIC_RUNTIME)" ] || [ "$(BUILD_SPECIFIC_RUNTIME)" == "ray" ]; then \
if [ -e Dockerfile.ray ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.ray \
TRANSFORM_RUNTIME_SRC_FILE=$(TRANSFORM_RAY_SRC) \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-ray \
BASE_IMAGE=$(RAY_BASE_IMAGE) \
test-image-sequence ; \
fi ;\
fi
@if [ -e Dockerfile.spark ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.spark \
TRANSFORM_RUNTIME_SRC_FILE=$(TRANSFORM_SPARK_SRC) \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-spark \
BASE_IMAGE=$(SPARK_BASE_IMAGE) \
test-image-sequence ; \
@if [ -z "$(BUILD_SPECIFIC_RUNTIME)" ] || [ "$(BUILD_SPECIFIC_RUNTIME)" == "spark" ]; then \
if [ -e Dockerfile.spark ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.spark \
TRANSFORM_RUNTIME_SRC_FILE=$(TRANSFORM_SPARK_SRC) \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-spark \
BASE_IMAGE=$(SPARK_BASE_IMAGE) \
test-image-sequence ; \
fi ;\
fi
-rm -rf data-processing-dist


image-python:
@if [ -e Dockerfile.python ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.python \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-python \
.defaults.lib-whl-image ; \
fi
$(MAKE) BUILD_SPECIFIC_RUNTIME=python image

image-ray:
@if [ -e Dockerfile.ray ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.ray \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-ray \
BASE_IMAGE=$(RAY_BASE_IMAGE) \
.defaults.lib-whl-image ; \
fi
$(MAKE) BUILD_SPECIFIC_RUNTIME=ray image

image-spark:
@if [ -e Dockerfile.spark ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.spark \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-spark \
BASE_IMAGE=$(SPARK_BASE_IMAGE) \
.defaults.lib-whl-image ; \
fi
$(MAKE) BUILD_SPECIFIC_RUNTIME=spark image

image:: .default.build-lib-wheel
## Build all possible images unless a specific runtime is specified
@if [ -z "$(BUILD_SPECIFIC_RUNTIME)" ] || [ "$(BUILD_SPECIFIC_RUNTIME)" == "python" ]; then \
$(MAKE) image-python ; \
if [ -e Dockerfile.python ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.python \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-python \
.defaults.lib-whl-image ; \
fi ; \
fi
@if [ -z "$(BUILD_SPECIFIC_RUNTIME)" ] || [ "$(BUILD_SPECIFIC_RUNTIME)" == "ray" ]; then \
$(MAKE) image-ray ; \
if [ -e Dockerfile.ray ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.ray \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-ray \
BASE_IMAGE=$(RAY_BASE_IMAGE) \
.defaults.lib-whl-image ; \
fi ; \
fi
@if [ -z "$(BUILD_SPECIFIC_RUNTIME)" ] || [ "$(BUILD_SPECIFIC_RUNTIME)" == "spark" ]; then \
$(MAKE) image-spark ; \
if [ -e Dockerfile.spark ]; then \
$(MAKE) DOCKER_FILE=Dockerfile.spark \
DOCKER_IMAGE_NAME=$(TRANSFORM_NAME)-spark \
BASE_IMAGE=$(SPARK_BASE_IMAGE) \
.defaults.lib-whl-image ; \
fi ; \
fi
-rm -rf data-processing-dist

Expand Down
16 changes: 16 additions & 0 deletions transforms/Makefile.transform.template
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
REPOROOT=../../..
# Use make help, to see the available rules
include $(REPOROOT)/transforms/.make.cicd.targets

#
# This is intended to be included across the Makefiles provided within
# a given transform's directory tree, so must use compatible syntax.
#
################################################################################
# This defines the name of the transform and is used to match against
# expected files and is used to define the transform's image name.
TRANSFORM_NAME=$(shell basename `pwd`)

################################################################################


Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ RUN useradd -ms /bin/bash dpk
USER dpk
WORKDIR /home/dpk

ARG PIP_INSTALL_EXTRA_ARGS
ARG DPK_WHEEL_FILE_NAME

# Copy and install data processing libraries
Expand All @@ -19,20 +20,9 @@ RUN pip install data-processing-dist/${DPK_WHEEL_FILE_NAME}

# END OF STEPS destined for a data-prep-kit base image

COPY --chown=dpk:root src/ src/
COPY --chown=dpk:root pyproject.toml pyproject.toml
COPY --chown=dpk:root dpk_doc_chunk/ dpk_doc_chunk/
COPY --chown=dpk:root requirements.txt requirements.txt
RUN pip install --no-cache-dir -e .

# copy transform main() entry point to the image
COPY ./src/html2parquet_transform_python.py .

# copy some of the samples in
COPY ./src/html2parquet_local.py local/

# copy test
COPY test/ test/
COPY test-data/ test-data/
RUN pip install ${PIP_INSTALL_EXTRA_ARGS} -r requirements.txt

# Set environment
ENV PYTHONPATH /home/dpk
Expand Down
27 changes: 27 additions & 0 deletions transforms/language/doc_chunk/Dockerfile.ray
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
ARG BASE_IMAGE=docker.io/rayproject/ray:2.24.0-py310
FROM ${BASE_IMAGE}

# install pytest
RUN pip install --no-cache-dir pytest

ARG PIP_INSTALL_EXTRA_ARGS
ARG DPK_WHEEL_FILE_NAME

# Copy and install data processing libraries
# These are expected to be placed in the docker context before this is run (see the make image).
COPY --chown=ray:users data-processing-dist data-processing-dist
RUN pip install data-processing-dist/${DPK_WHEEL_FILE_NAME}[ray]


COPY --chown=ray:users dpk_doc_chunk/ dpk_doc_chunk/
COPY --chown=ray:users requirements.txt requirements.txt
RUN pip install ${PIP_INSTALL_EXTRA_ARGS} --no-cache-dir -r requirements.txt

# Set environment
ENV PYTHONPATH /home/ray

# Put these at the end since they seem to upset the docker cache.
ARG BUILD_DATE
ARG GIT_COMMIT
LABEL build-date=$BUILD_DATE
LABEL git-commit=$GIT_COMMIT
99 changes: 22 additions & 77 deletions transforms/language/doc_chunk/Makefile
Original file line number Diff line number Diff line change
@@ -1,79 +1,24 @@
REPOROOT=../../..
# Use make help, to see the available rules
include $(REPOROOT)/.make.defaults

setup::
@# Help: Recursively make $@ all subdirs
$(MAKE) RULE=$@ .recurse

clean::
@# Help: Recursively make $@ all subdirs
$(MAKE) RULE=$@ .recurse

build::
@# Help: Recursively make $@ in subdirs
$(MAKE) RULE=$@ .recurse
venv::
@# Help: Recursively make $@ in subdirs
$(MAKE) RULE=$@ .recurse

image::
@# Help: Recursively make $@ in all subdirs
@$(MAKE) RULE=$@ .recurse

set-versions:
@# Help: Recursively $@ in all subdirs
@$(MAKE) RULE=$@ .recurse

publish::
@# Help: Recursively make $@ in all subdirs
@$(MAKE) RULE=$@ .recurse

test-image::
@# Help: Recursively make $@ in all subdirs
@$(MAKE) RULE=$@ .recurse

test::
@# Help: Recursively make $@ in all subdirs
@$(MAKE) RULE=$@ .recurse

test-src::
@# Help: Recursively make $@ in all subdirs
$(MAKE) RULE=$@ .recurse

kind-load-image::
@# Help: Recursively make $@ in all subdirs
$(MAKE) RULE=$@ .recurse

docker-load-image::
@# Help: Recursively make $@ in all subdirs
$(MAKE) RULE=$@ .recurse

docker-save-image::
@# Help: Recursively make $@ in all subdirs
$(MAKE) RULE=$@ .recurse

.PHONY: workflow-venv
workflow-venv:
if [ -e kfp_ray ]; then \
$(MAKE) -C kfp_ray workflow-venv; \
fi

.PHONY: workflow-test
workflow-test:
if [ -e kfp_ray ]; then \
$(MAKE) -C kfp_ray workflow-test; \
fi

.PHONY: workflow-upload
workflow-upload:
if [ -e kfp_ray ]; then \
$(MAKE) -C kfp_ray workflow-upload; \
fi

.PHONY: workflow-build
workflow-build:
if [ -e kfp_ray ]; then \
$(MAKE) -C kfp_ray workflow-build; \
fi

include $(REPOROOT)/transforms/.make.cicd.targets

#
# This is intended to be included across the Makefiles provided within
# a given transform's directory tree, so must use compatible syntax.
#
################################################################################
# This defines the name of the transform and is used to match against
# expected files and is used to define the transform's image name.
TRANSFORM_NAME=$(shell basename `pwd`)

################################################################################


LINUX_WITH_CPU_TORCH?=true
OS := $(shell uname -s)
ifeq ($(OS),Linux)
ifeq ($(LINUX_WITH_CPU_TORCH),true)
PIP_INSTALL_EXTRA_ARGS=--extra-index-url=https://download.pytorch.org/whl/cpu
DOCKER_BUILD_EXTRA_ARGS=--build-arg PIP_INSTALL_EXTRA_ARGS=${PIP_INSTALL_EXTRA_ARGS}
endif
endif
Loading
Loading