🐛 Fix data type

NONONOexe · Oct 7, 2024 · 9f9975e · 9f9975e
1 parent 3b1c131
commit 9f9975e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/R/post-process.R b/R/post-process.R
@@ -52,13 +52,6 @@ post_process_main <- function(data) {
   accident_data$latitude  <- convert_deg(accident_data$latitude)
   accident_data$longitude <- convert_deg(accident_data$longitude)
 
-  # Convert to sf object
-  accident_data_sf <- st_as_sf(
-    accident_data,
-    coords = c("longitude", "latitude"),
-    crs = 4326
-  )
-
   # Filter person data
   filter_person_data <- function(data, tag, suf) {
     person_data <- filter_data_by_tag(data, tag)
@@ -74,7 +67,11 @@ post_process_main <- function(data) {
 
   # Combine process data
   processed_data <- list(
-    accident = cbind(key_data, accident_data_sf),
+    accident = st_as_sf(
+      cbind(key_data, accident_data),
+      coords = c("longitude", "latitude"),
+      crs = 4326
+    ),
     person   = rbind(
       cbind(key_data, person_a_data),
       cbind(key_data, person_b_data)

diff --git a/R/read-accident-data.R b/R/read-accident-data.R
@@ -37,7 +37,7 @@ read_accident_data <- function(
 
   # Add attributes and class
   attr(accident_data, "dataset_name") <- dataset_name
-  class(accident_data) <- c("accident_data", "data.frame")
+  class(accident_data) <- c("accident_data", class(accident_data))
 
   # Apply post-processing if provided
   if (is.function(post_process)) {