...

1 files changed, 7 insertions, 0 deletions

diff --git a/tnet_io/init.c b/tnet_io/init.c
index 3fa7cb8..16f6f37 100644
--- a/tnet_io/init.c
+++ b/tnet_io/init.c
@@ -40,10 +40,17 @@ static int feat_repo_next(lua_State *L) {
     return 0;
 }
 
+static int feat_repo_is_end(lua_State *L) {
+    TNetFeatureRepo *repo = luaT_checkudata(L, 1, nerv_tnet_feat_repo_tname);
+    lua_pushboolean(L, tnet_feature_repo_is_end(repo));
+    return 1;
+}
+
 static const luaL_Reg feat_repo_methods[] = {
     {"cur_utter", feat_repo_current_utterance},
     {"cur_tag", feat_repo_current_tag},
     {"next", feat_repo_next},
+    {"is_end", feat_repo_is_end},
     {NULL, NULL}
 };