Skip to content

Commit a0afd87

Browse files
committed
feat: add argument for mindepth
1 parent d488698 commit a0afd87

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

src/wrapper.rs

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -320,6 +320,24 @@ impl PyNutsSettings {
320320
}
321321
}
322322

323+
#[getter]
324+
fn mindepth(&self) -> u64 {
325+
match &self.inner {
326+
Settings::Diag(nuts_settings) => nuts_settings.mindepth,
327+
Settings::LowRank(nuts_settings) => nuts_settings.mindepth,
328+
Settings::Transforming(nuts_settings) => nuts_settings.mindepth,
329+
}
330+
}
331+
332+
#[setter(maxdepth)]
333+
fn set_mindepth(&mut self, val: u64) {
334+
match &mut self.inner {
335+
Settings::Diag(nuts_settings) => nuts_settings.mindepth = val,
336+
Settings::LowRank(nuts_settings) => nuts_settings.mindepth = val,
337+
Settings::Transforming(nuts_settings) => nuts_settings.mindepth = val,
338+
}
339+
}
340+
323341
#[getter]
324342
fn store_gradient(&self) -> bool {
325343
match &self.inner {

0 commit comments

Comments
 (0)