lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove Keras Core refs

Browse Source
This commit is contained in:
Neel Kovelamudi 2023-09-25 16:41:01 +00:00 committed by François Chollet
parent 2f5dd2fc98
commit 6b38644fc6
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
integration_tests/import_test.py
View File

@ -2,7 +2,7 @@ import os
import re import re
import subprocess import subprocess
from keras_core import backend from keras import backend
BACKEND_REQ = { BACKEND_REQ = {
"tensorflow": "tensorflow", "tensorflow": "tensorflow",
@ -25,7 +25,7 @@ def setup_package():
build_process.stdout, build_process.stdout,
) )
if not match: if not match:
raise ValueError("Installing Keras Core package unsuccessful. ") raise ValueError("Installing Keras package unsuccessful. ")
print(build_process.stderr) print(build_process.stderr)
whl_path = match.group() whl_path = match.group()
return whl_path return whl_path
@ -61,7 +61,7 @@ def manage_venv_installs(whl_path):
run_commands_venv(install_setup) run_commands_venv(install_setup)
def run_keras_core_flow(): def run_keras_flow():
test_script = [ test_script = [
# Runs the example script # Runs the example script
"python -m pytest integration_tests/basic_full_flow.py", "python -m pytest integration_tests/basic_full_flow.py",
@ -94,7 +94,7 @@ def run_commands_venv(commands):
p.wait() p.wait()
def test_keras_core_imports(): def test_keras_imports():
# Ensures packages from all backends are installed. # Ensures packages from all backends are installed.
# Builds Keras core package and returns package file path. # Builds Keras core package and returns package file path.
whl_path = setup_package() whl_path = setup_package()
@ -108,11 +108,11 @@ def test_keras_core_imports():
# Runs test of basic flow in Keras Core. # Runs test of basic flow in Keras Core.
# Tests for backend-specific imports and `model.fit()`. # Tests for backend-specific imports and `model.fit()`.
run_keras_core_flow() run_keras_flow()
# Removes virtual environment and associated files # Removes virtual environment and associated files
cleanup() cleanup()
if __name__ == "__main__": if __name__ == "__main__":
test_keras_core_imports() test_keras_imports()