fix test_auto_unit (#1035)

Summary:
Pull Request resolved: #1035

as title. There's another triton error message which was introduced at some point

Reviewed By: diego-urgell

Differential Revision: D83592092

fbshipit-source-id: 00c74d56a8a62c77192f441d7799fdb617c2d4c1

Author: galrotem

tests/framework/test_auto_unit.py

@@ -379,10 +379,10 @@ def configure_model(self):
             ):
                 unit_type(module=module)
 
-                expected_warnings = [
+                self.assertTrue(
                     "The self.module attribute is managed by AutoUnit and is not meant to be reassigned."
-                ]
-                self.assertEqual(warning_container, expected_warnings)
+                    in warning_container
+                )
 
     @skip_if_not_distributed
     def test_auto_unit_ddp(self) -> None: