Fix ddp argument bug

pomonam · Mar 19, 2024 · 630affa · 630affa
1 parent 6beba81
commit 630affa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/gpu_tests/ddp_variation_test.py b/tests/gpu_tests/ddp_variation_test.py
@@ -115,8 +115,8 @@ def test_lambda_matrices(self):
             for mp in module_partition_lst:
                 for ca in cached_activation_cpu_offload:
                     factor_args = FactorArguments(
-                        covariance_data_partition_size=dp,
-                        covariance_module_partition_size=mp,
+                        lambda_data_partition_size=dp,
+                        lambda_module_partition_size=mp,
                         cached_activation_cpu_offload=ca,
                     )
                     self.analyzer.fit_lambda_matrices(