Remove Keras Core refs
This commit is contained in:
parent
2f5dd2fc98
commit
6b38644fc6
@ -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()
|
Loading…
Reference in New Issue
Block a user