Skip to content

Commit e979226

Browse files
committed
feat: expose static trajectory length in nuts
1 parent 76439f3 commit e979226

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

src/wrapper.rs

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -629,6 +629,31 @@ impl PyNutsSettings {
629629
}
630630
Ok(())
631631
}
632+
633+
#[getter]
634+
fn check_turning(&self) -> Result<bool> {
635+
match &self.inner {
636+
Settings::LowRank(inner) => Ok(inner.check_turning),
637+
Settings::Diag(inner) => Ok(inner.check_turning),
638+
Settings::Transforming(inner) => Ok(inner.check_turning),
639+
}
640+
}
641+
642+
#[setter(check_turning)]
643+
fn set_check_turning(&mut self, val: bool) -> Result<()> {
644+
match &mut self.inner {
645+
Settings::LowRank(inner) => {
646+
inner.check_turning = val;
647+
}
648+
Settings::Diag(inner) => {
649+
inner.check_turning = val;
650+
}
651+
Settings::Transforming(inner) => {
652+
inner.check_turning = val;
653+
}
654+
}
655+
Ok(())
656+
}
632657
}
633658

634659
pub(crate) enum SamplerState {

0 commit comments

Comments
 (0)