update tests

Author: ahosler

tests/operators/anomaly/test_anomaly_simple.py

@@ -52,7 +52,7 @@
         parameters_short.append((m, d))
 
 
-@pytest.mark.parametrize("model", MODELS)
+@pytest.mark.parametrize("model", ["autots"])
 def test_artificial_big(model):
     all_data = []
     TARGET_COLUMN = "sensor"

tests/unitary/with_extras/langchain/test_guardrails.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*--
 
-# Copyright (c) 2023 Oracle and/or its affiliates.
+# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import json
@@ -151,7 +151,9 @@ def test_fn(chain: GuardrailSequence):
             self.assertIsInstance(output.data, str)
             self.assertEqual(output.data, "Tell me a joke about cats")
             self.assertIsInstance(output.info, list)
-            self.assertEqual(len(output.info), len(chain.steps))
+            self.assertEqual(
+                len(output.info), len(chain.steps) if hasattr(chain, "steps") else 0
+            )
 
         self.assert_before_and_after_serialization(test_fn, chain)
 
@@ -175,7 +177,6 @@ def test_fn(chain: GuardrailSequence):
 
         self.assert_before_and_after_serialization(test_fn, chain)
 
-
     def test_save_to_file(self):
         """Tests saving to file."""
         message = "Let's talk something else."

tests/unitary/with_extras/langchain/test_serialization.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*--
 
-# Copyright (c) 2023 Oracle and/or its affiliates.
+# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 
@@ -142,7 +142,6 @@ def test_llm_chain_serialization_with_oci(self):
         self.assertEqual(llm_chain.llm.model, "my_model")
         self.assertEqual(llm_chain.input_keys, ["subject"])
 
-
     def test_oci_gen_ai_serialization(self):
         """Tests serialization of OCI Gen AI LLM."""
         try:
@@ -189,7 +188,7 @@ def test_runnable_sequence_serialization(self):
 
         element_1 = kwargs.get("first")
         self.assertEqual(element_1.get("_type"), "RunnableParallel")
-        step = element_1.get("kwargs").get("steps").get("text")
+        step = element_1.get("kwargs", dict()).get("steps", dict()).get("text", dict())
         self.assertEqual(step.get("id")[-1], "RunnablePassthrough")
 
         element_2 = kwargs.get("middle")[0]