Skip to content

Commit 2ae5c5f

Browse files
committed
fix CiteEntry import
1 parent a0e92d1 commit 2ae5c5f

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

bioimageio/core/build_spec/build_model.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import hashlib
33
import os
44
from pathlib import Path
5-
from shutil import copyfile
65
from typing import Any, Dict, List, Optional, Tuple, Union
76

87
import imageio
@@ -14,8 +13,7 @@
1413
from bioimageio.core import export_resource_package, load_raw_resource_description
1514
from bioimageio.core.resource_io.nodes import URI
1615
from bioimageio.core.resource_io.utils import resolve_local_source, resolve_source
17-
from bioimageio.spec.shared import fields
18-
from bioimageio.spec.shared.raw_nodes import ImportableSourceFile, ImportableModule
16+
from bioimageio.spec.shared.raw_nodes import ImportableModule, ImportableSourceFile
1917

2018
try:
2119
from typing import get_args
@@ -217,7 +215,7 @@ def _get_output_tensor(path, name, reference_tensor, scale, offset, axes, data_r
217215

218216
# TODO The citation entry should be improved so that we can properly derive doi vs. url
219217
def _build_cite(cite: Dict[str, str]):
220-
citation_list = [model_spec.raw_nodes.CiteEntry(text=k, url=v) for k, v in cite.items()]
218+
citation_list = [spec.rdf.raw_nodes.CiteEntry(text=k, url=v) for k, v in cite.items()]
221219
return citation_list
222220

223221

0 commit comments

Comments
 (0)