Compare attributes.

tests/test_metalearner.py

@@ -1125,12 +1125,16 @@ def test_validate_outcome_different_classes(implementation, use_pandas, rng):
     "implementation",
     [TLearner, SLearner, XLearner, RLearner, DRLearner],
 )
-def test_init_args_smoke(implementation):
+def test_init_args(implementation):
     ml = implementation(
         True,
         2,
         LogisticRegression,
         LinearRegression,
         LogisticRegression,
     )
-    implementation(**ml.init_args)
+    ml2 = implementation(**ml.init_args)
+
+    assert set(ml.__dict__.keys()) == set(ml2.__dict__.keys())
+    for key in ml.__dict__:
+        assert ml.__dict__[key] == ml2.__dict__[key]