Skip to content

Commit

Permalink
Bug Fix 2
Browse files Browse the repository at this point in the history
  • Loading branch information
FBerding committed Oct 9, 2023
1 parent dda544b commit 51bdbcd
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 1 addition & 2 deletions tests/testthat/test-04_transformer_models.R
Original file line number Diff line number Diff line change
Expand Up @@ -654,6 +654,7 @@ for(ai_method in ai_methods){
expect_no_error(bert_modeling$get_bow_components())
})

#------------------------------------------------------------------------
test_that(paste0(ai_method,"Save Total Model H5",framework), {
expect_no_error(
save_ai_model(model=bert_modeling,
Expand Down Expand Up @@ -693,7 +694,6 @@ for(ai_method in ai_methods){

#-------------------------------------------------------------------------
test_that(paste0(ai_method,"Save Total Model TF without ID",framework), {
testthat::skip_on_os("linux")
expect_no_error(
save_ai_model(model=bert_modeling,
model_dir = testthat::test_path(paste0(path_03,"/",framework)),
Expand All @@ -703,7 +703,6 @@ for(ai_method in ai_methods){
})

test_that(paste0("Load Total Model TF without ID",framework), {
testthat::skip_on_os("linux")
bert_modeling<-NULL
bert_modeling<-load_ai_model(
ml_framework = framework,
Expand Down
2 changes: 2 additions & 0 deletions tests/testthat/test-14_classifier_neural_net.R
Original file line number Diff line number Diff line change
Expand Up @@ -569,6 +569,7 @@ test_that(paste(ml_framework,"Classifier Load Total Model H5"), {
base::gc(verbose = FALSE,full = TRUE)

test_that(paste(ml_framework,"Classifier Save Total Model TF with ID"), {
testthat::skip_on_os("linux")
expect_no_error(
save_ai_model(model=classifier,
model_dir = testthat::test_path("test_artefacts/tmp_full_models"),
Expand All @@ -578,6 +579,7 @@ test_that(paste(ml_framework,"Classifier Save Total Model TF with ID"), {
base::gc(verbose = FALSE,full = TRUE)

test_that(paste(ml_framework,"Classifier Load Total Model TF with ID"), {
testthat::skip_on_os("linux")
new_classifier<-NULL
new_classifier<-load_ai_model(
ml_framework="tensorflow",
Expand Down

0 comments on commit 51bdbcd

Please sign in to comment.