Skip to content

Commit

Permalink
Merge pull request #48 from franneck94/fixTest
Browse files Browse the repository at this point in the history
Fix test
  • Loading branch information
franneck94 authored May 20, 2024
2 parents 66f5b62 + 0e1830d commit 217d75e
Show file tree
Hide file tree
Showing 11 changed files with 20 additions and 10 deletions.
4 changes: 4 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,9 @@
# TensorCross Change Log

## Version 2.0.0: May 20, 2024

- This will be the stable version for TF 2.13+ which is the first version that supports keras 3.0

## Version 1.0.0: October 29, 2023

- Some rework, updates docs and versioning. This will be the stable version for TF (Keras) 2.8+
Expand Down
2 changes: 1 addition & 1 deletion examples/example_grid_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=1)
x_input = Input(shape=(1,))
y_pred = Dense(units=1)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
2 changes: 1 addition & 1 deletion examples/example_grid_search_cv.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=1)
x_input = Input(shape=(1,))
y_pred = Dense(units=1)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
2 changes: 1 addition & 1 deletion examples/example_random_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=1)
x_input = Input(shape=(1,))
y_pred = Dense(units=1)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
2 changes: 1 addition & 1 deletion examples/example_random_search_cv.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=1)
x_input = Input(shape=(1,))
y_pred = Dense(units=1)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
8 changes: 7 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,13 @@
ISRELEASED = False

MIN_PYTHON_VERSION = "3.9"
INSTALL_REQUIRES = ["keras>=2.8", "numpy", "scipy", "scikit-learn>=1.0"]
INSTALL_REQUIRES = [
"tensorflow>=2.13",
"keras>=3.0",
"numpy",
"scipy",
"scikit-learn>=1.0",
]


PACKAGES = find_packages(include=["tensorcross", "tensorcross.*"])
Expand Down
2 changes: 1 addition & 1 deletion tensorcross/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "1.0.0"
__version__ = "2.0.0"
2 changes: 1 addition & 1 deletion tests/test_grid_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=num_features)
x_input = Input(shape=(num_features,))
y_pred = Dense(units=num_targets)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
2 changes: 1 addition & 1 deletion tests/test_grid_search_cv.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=num_features)
x_input = Input(shape=(num_features,))
y_pred = Dense(units=num_targets)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
2 changes: 1 addition & 1 deletion tests/test_random_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=num_features)
x_input = Input(shape=(num_features,))
y_pred = Dense(units=num_targets)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down
2 changes: 1 addition & 1 deletion tests/test_random_search_cv.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ def build_model(
learning_rate: float,
) -> Model:
"""Build the test model."""
x_input = Input(shape=num_features)
x_input = Input(shape=(num_features,))
y_pred = Dense(units=num_targets)(x_input)
model = Model(inputs=[x_input], outputs=[y_pred])

Expand Down

0 comments on commit 217d75e

Please sign in to comment.