Skip to content

Commit 8995b40

Browse files
Update requirements.txt and pyproject.toml (#40)
* Update `requirements.txt` and `pyproject.toml` * Update image url * Minor fix
1 parent f6ef9f3 commit 8995b40

File tree

4 files changed

+8
-5
lines changed

4 files changed

+8
-5
lines changed

kimm/models/base_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ def get_weights_url(self, weights):
225225

226226
for _weights, _origin, _file_name in self.available_weights:
227227
if weights == _weights:
228-
return f"{_origin}/{_file_name}"
228+
return f"{_origin}{_file_name}"
229229

230230
# Failed to find the weights
231231
_available_weights_name = [

kimm/models/models_test.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,8 @@ def test_model_base_channels_last(
378378
backend.set_image_data_format("channels_last")
379379
model = model_class(weights=weights)
380380
image_path = keras.utils.get_file(
381-
"african_elephant.jpg", "https://i.imgur.com/Bvro0YD.png"
381+
"elephant.png",
382+
"https://github.com/james77777778/keras-image-models/releases/download/0.1.0/elephant.png",
382383
)
383384
# preprocessing
384385
image = utils.load_img(image_path, target_size=(image_size, image_size))
@@ -411,7 +412,8 @@ def test_model_base_channels_first(
411412
backend.set_image_data_format("channels_first")
412413
model = model_class(weights=weights)
413414
image_path = keras.utils.get_file(
414-
"african_elephant.jpg", "https://i.imgur.com/Bvro0YD.png"
415+
"elephant.png",
416+
"https://github.com/james77777778/keras-image-models/releases/download/0.1.0/elephant.png",
415417
)
416418
# preprocessing
417419
image = utils.load_img(image_path, target_size=(image_size, image_size))

pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,14 @@ authors = [{ name = "Hong-Yu Chiu", email = "james77777778@gmail.com" }]
2020
maintainers = [{ name = "Hong-Yu Chiu", email = "james77777778@gmail.com" }]
2121
readme = "README.md"
2222
requires-python = ">=3.9"
23-
license = { file = "LICENSE" }
23+
license = { text = "Apache License 2.0" }
2424
classifiers = [
2525
"Programming Language :: Python",
2626
"Programming Language :: Python :: 3",
2727
"Programming Language :: Python :: 3.9",
2828
"Programming Language :: Python :: 3.10",
2929
"Programming Language :: Python :: 3.11",
30+
"Programming Language :: Python :: 3.12",
3031
"Programming Language :: Python :: 3 :: Only",
3132
"Operating System :: Unix",
3233
"Operating System :: MacOS",

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
# "jax[cuda12_local]"
1111

1212
# Following is for github runner
13-
tf-nightly-cpu==2.16.0.dev20240101
13+
tensorflow-cpu==2.16.0rc0 # TODO: wait for TF 2.16 release
1414

1515
--extra-index-url https://download.pytorch.org/whl/cpu
1616
torch>=2.1.0

0 commit comments

Comments
 (0)