feat: Set output prefix
This commit is contained in:
parent
e02ec4d257
commit
077651e708
12
nhf/build.py
12
nhf/build.py
|
@ -214,7 +214,7 @@ class Submodel:
|
||||||
def write_to(self, obj, path: str):
|
def write_to(self, obj, path: str):
|
||||||
x = self._method(obj)
|
x = self._method(obj)
|
||||||
assert isinstance(x, Model), f"Unexpected type: {type(x)}"
|
assert isinstance(x, Model), f"Unexpected type: {type(x)}"
|
||||||
x.build_all(path)
|
x.build_all(path, prefix=False)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def methods(cls, subject):
|
def methods(cls, subject):
|
||||||
|
@ -271,11 +271,17 @@ class Model:
|
||||||
total += 1
|
total += 1
|
||||||
return total
|
return total
|
||||||
|
|
||||||
def build_all(self, output_dir: Union[Path, str] = "build", verbose=1):
|
def build_all(
|
||||||
|
self,
|
||||||
|
output_dir: Union[Path, str] = "build",
|
||||||
|
prefix: bool = True,
|
||||||
|
verbose=1):
|
||||||
"""
|
"""
|
||||||
Build all targets in this model and write the results to file
|
Build all targets in this model and write the results to file
|
||||||
"""
|
"""
|
||||||
output_dir = Path(output_dir) / self.name
|
output_dir = Path(output_dir)
|
||||||
|
if prefix:
|
||||||
|
output_dir = output_dir / self.name
|
||||||
targets = Target.methods(self)
|
targets = Target.methods(self)
|
||||||
for t in targets.values():
|
for t in targets.values():
|
||||||
file_name = t.file_name
|
file_name = t.file_name
|
||||||
|
|
Loading…
Reference in New Issue