mirror of
https://github.com/luxfhe/ml-sdk.git
synced 2026-10-30 18:13:04 +00:00
zeekay
bde2e2531b
Machine Learning with FHE - fork of Zama Concrete-ML - Built-in FHE-friendly ML models - scikit-learn compatible API - PyTorch model conversion - Quantization-aware training
58 lines
1.2 KiB
Bash
Executable File
58 lines
1.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -e
|
|
|
|
TEST_DIR="/tmp/cml_codeblocks"
|
|
rm -rf "${TEST_DIR}"
|
|
mkdir -p "${TEST_DIR}"
|
|
|
|
# grep -v "^\./\." is to avoid files in .hidden_directories
|
|
# grep -v "api/concrete\.ml" is to avoid autogen API doc since lazydocs produces bad python blocks
|
|
MD_FILES=$(find . -type f -name "*.md" | grep -v "^\./\." | grep -v "api/concrete\.ml")
|
|
|
|
# Force NCPU to 1, since using parallel checks makes issues on linux or CI
|
|
NCPU=1
|
|
|
|
while [ -n "$1" ]
|
|
do
|
|
case "$1" in
|
|
"--file" )
|
|
shift
|
|
MD_FILES="$1"
|
|
NCPU=1
|
|
;;
|
|
|
|
*)
|
|
echo "Unknown param : $1"
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
DEST_MDS=()
|
|
|
|
for MD_FILE in $MD_FILES
|
|
do
|
|
DEST_MD="${TEST_DIR}/${MD_FILE}"
|
|
NEW_DIR=$(dirname "${DEST_MD}")
|
|
mkdir -p "$NEW_DIR"
|
|
cp "${MD_FILE}" "${DEST_MD}"
|
|
DEST_MDS+=("${DEST_MD}")
|
|
done
|
|
|
|
poetry run python ./script/make_utils/deactivate_docs_admonitions_for_tests.py \
|
|
--files "${DEST_MDS[@]}"
|
|
|
|
set -x
|
|
|
|
make pytest_internal_parallel TEST="${TEST_DIR}" N_CPU="${NCPU}" PYTEST_OPTIONS="--codeblocks"
|
|
|
|
set +x
|
|
|
|
rm -rf "${TEST_DIR}"
|
|
|
|
# Remove onnx files created while running the codeblocks
|
|
rm -f "tmp.model.onnx"
|
|
rm -f "tmp.onnx"
|