diff options
-rw-r--r-- | python/sentencepiece.i | 4 | ||||
-rw-r--r-- | python/sentencepiece_wrap.cxx | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/python/sentencepiece.i b/python/sentencepiece.i index 784411b..dd2fea0 100644 --- a/python/sentencepiece.i +++ b/python/sentencepiece.i @@ -106,7 +106,7 @@ class PySentenceIterator : public sentencepiece::SentenceIterator { } ~PySentenceIterator() { - if (iter_) Py_DECREF(iter_); + Py_XDECREF(iter_); } bool done() const override { @@ -144,7 +144,7 @@ class PySentenceIterator : public sentencepiece::SentenceIterator { status_ = sentencepiece::util::Status(sentencepiece::util::StatusCode::kInternal, "Not a string."); } - Py_DECREF(item_); + Py_XDECREF(item_); } PyObject *iter_ = nullptr; PyObject *item_ = nullptr; diff --git a/python/sentencepiece_wrap.cxx b/python/sentencepiece_wrap.cxx index 3bced05..bd85063 100644 --- a/python/sentencepiece_wrap.cxx +++ b/python/sentencepiece_wrap.cxx @@ -2877,7 +2877,7 @@ class PySentenceIterator : public sentencepiece::SentenceIterator { } ~PySentenceIterator() { - if (iter_) Py_DECREF(iter_); + Py_XDECREF(iter_); } bool done() const override { @@ -2915,7 +2915,7 @@ class PySentenceIterator : public sentencepiece::SentenceIterator { status_ = sentencepiece::util::Status(sentencepiece::util::StatusCode::kInternal, "Not a string."); } - Py_DECREF(item_); + Py_XDECREF(item_); } PyObject *iter_ = nullptr; PyObject *item_ = nullptr; |