default 0 for nth

h2oai · Mar 12, 2023 · 21226cc · 21226cc
1 parent 7872d45
commit 21226cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/core/expr/fexpr.cc b/src/core/expr/fexpr.cc
@@ -534,8 +534,7 @@ DECLARE_METHOD(&PyFExpr::min)
 
 oobj PyFExpr::nth(const XArgs& args) {
   auto nthFn = oobj::import("datatable", "nth");
-  oobj n = args[0].to_oobj() ? args[0].to_oobj() 
-                             : py::oint(0);
+  auto n = args[1].to<py::oobj>(py::oint(0));
   oobj skipna = args[1].to_oobj_or_none();
   return nthFn.call({this, n, skipna});
 }

diff --git a/src/core/expr/fexpr_nth.cc b/src/core/expr/fexpr_nth.cc
@@ -110,7 +110,7 @@ class FExpr_Nth : public FExpr_Func {
 
 static py::oobj pyfn_nth(const py::XArgs& args) {
   auto arg = args[0].to_oobj();
-  auto n = args[1].to_oobj();
+  auto n = args[1].to<py::oobj>(py::oint(0));
   auto skipna = args[2].to<bool>(false);
   if (!n.is_int()) {
     throw TypeError() << "The argument for the `nth` parameter "