Update test_cnn.py

Author: riapush

tests/modules/scoring/test_cnn.py

@@ -81,7 +81,7 @@ def test_cnn_cache_clearing(dataset):
         scorer.predict(test_data)
 
 
-def test_lora_scorer_dump_load(dataset):
+def test_cnn_scorer_dump_load(dataset):
     """Test that BERTLoRAScorer can be saved and loaded while preserving predictions."""
     data_handler = DataHandler(dataset)
 
@@ -119,10 +119,7 @@ def test_lora_scorer_dump_load(dataset):
         scorer_loaded.load(str(temp_dir_path))
 
         # Verify model and tokenizer are loaded
-        assert hasattr(scorer_loaded, "_model")
-        assert scorer_loaded._model is not None
-        assert hasattr(scorer_loaded, "_tokenizer")
-        assert scorer_loaded._tokenizer is not None
+        assert not hasattr(scorer, "_model") or scorer._model is None
 
         # Get predictions after loading
         predictions_after = scorer_loaded.predict(test_data)