fix fexpr.cc

samukweku · samukweku · commit 7435bf5d9938 · 2023-04-11T21:59:20.000+10:00
diff --git a/src/core/expr/fexpr.cc b/src/core/expr/fexpr.cc
@@ -533,7 +533,8 @@ DECLARE_METHOD(&PyFExpr::min)
 
 oobj PyFExpr::nth(const XArgs& args) {
   auto nthFn = oobj::import("datatable", "nth");
-  auto n = args[1].to<py::oobj>(py::oint(0));
+  oobj n = args[0].to_oobj() ? args[0].to_oobj() 
+                             : py::oint(0);
   oobj skipna = args[1].to_oobj_or_none();
   return nthFn.call({this, n, skipna});
 }

Original file line number	Diff line number	Diff line change
`@@ -533,7 +533,8 @@ DECLARE_METHOD(&PyFExpr::min)`
`533`	`533`
`534`	`534`	`oobj PyFExpr::nth(const XArgs& args) {`
`535`	`535`	`auto nthFn = oobj::import("datatable", "nth");`
`536`		`- auto n = args[1].to<py::oobj>(py::oint(0));`
	`536`	`+ oobj n = args[0].to_oobj() ? args[0].to_oobj()`
	`537`	`+ : py::oint(0);`
`537`	`538`	`oobj skipna = args[1].to_oobj_or_none();`
`538`	`539`	`return nthFn.call({this, n, skipna});`
`539`	`540`	`}`