feature: print version
- fix: default config wasn't being selected for custom modelspull/172/head
parent
9b1d130f93
commit
5b56f05da7
@ -0,0 +1,6 @@
|
||||
from importlib.metadata import PackageNotFoundError, version
|
||||
|
||||
try:
|
||||
__version__ = version("imaginairy")
|
||||
except PackageNotFoundError:
|
||||
__version__ = None
|
@ -0,0 +1,16 @@
|
||||
from imaginairy import config
|
||||
from imaginairy.model_manager import resolve_model_paths
|
||||
|
||||
|
||||
def test_resolved_paths():
|
||||
"""Test that the resolved model path is correct."""
|
||||
model_metadata, weights_path, config_path = resolve_model_paths()
|
||||
assert model_metadata.short_name == config.DEFAULT_MODEL
|
||||
assert model_metadata.config_path == config_path
|
||||
default_config_path = config_path
|
||||
|
||||
model_metadata, weights_path, config_path = resolve_model_paths(
|
||||
weights_path="foo.ckpt"
|
||||
)
|
||||
assert weights_path == "foo.ckpt"
|
||||
assert config_path == default_config_path
|
Loading…
Reference in New Issue