minor changes to entity_types_map

PolMine · Apr 18, 2024 · c5786a5 · c5786a5
1 parent ae46578
commit c5786a5
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 8 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -2,7 +2,7 @@ Package: dbpedia
 Type: Package
 Title: R Wrapper for DBpedia Spotlight
 Version: 0.1.2.9006
-Date: 2024-04-16
+Date: 2024-04-18
 Authors@R: c(
   person("Andreas", "Blaette", role = c("aut", "cre"), email = "andreas.blaette@uni-due.de", comment = c(ORCID = "0000-0001-8970-8010")),
   person("Christoph", "Leonhardt", role = "aut")

diff --git a/R/entity_types.R b/R/entity_types.R
@@ -43,7 +43,7 @@ setGeneric(
 #'   corpus_reshape(to = "paragraphs")
 #'   
 #' uritab_paragraphs <- get_dbpedia_uris(
-#'   x = inaugural_paragraphs,
+#'   x = inaugural_paragraphs[2], # single paragraph
 #'   language = "en",
 #'   max_len = 5600L,
 #'   confidence = 0.5,
@@ -58,12 +58,12 @@ setGeneric(
 #'   "LOCATION" = "Place"
 #' )
 #'
-#' entity_types_map(
+#' categories <- entity_types_map(
 #'   uritab_paragraphs[["DBpedia_type"]],
 #'   mapping_vector = mapping_vector_character
 #' )
 #'
-#' # For the entire data.table and mulitple types sources:
+#' # For the entire data.table and multiple types sources:
 #'
 #' mapping_vector_dt = c(
 #'   "PERSON" = "DBpedia:Person",
@@ -112,6 +112,8 @@ setMethod(
       return(category)
     }
     )
+
+    return(retval)
   }
 )
 

diff --git a/man/entity_types_map.Rd b/man/entity_types_map.Rd
diff --git a/tests/testthat/test-entity_types_map.R b/tests/testthat/test-entity_types_map.R
@@ -166,7 +166,7 @@ test_that(
 
     # Examples based on processing quanteda's data_corpus_inaugural with the
     # English DBpedia Spotlight endpoint (entities occur in 2013-Obama).
-    
+
     types_vector <- c("|Species|Person|Eukaryote|Animal|Politician|",
                       "|Organisation|Agent|Legislature|",
                       "|Place|Location|CelestialBody|Planet|",