chore: lint code

Author: Sangamesh26

src/llama_prompt_ops/interfaces/cli.py

@@ -799,7 +799,7 @@ def migrate(config, model, output_dir, save_yaml, api_key_env, dotenv_path, log_
     except ValueError as e:
         click.echo(f"Error: {str(e)}", err=True)
         sys.exit(1)
-    
+
     # Validate the minimum number of records in dataset
     try:
         validate_min_records_in_dataset(dataset_adapter)

tests/integration/test_cli_integration.py

@@ -131,7 +131,10 @@ def test_cli_migrate_command(self, mock_api_key_check, temp_config_file):
                 "llama_prompt_ops.interfaces.cli.get_strategy", return_value=MagicMock()
             ),
             patch("llama_prompt_ops.interfaces.cli.load_config", return_value={}),
-            patch("llama_prompt_ops.interfaces.cli.validate_min_records_in_dataset", return_value=None),
+            patch(
+                "llama_prompt_ops.interfaces.cli.validate_min_records_in_dataset",
+                return_value=None,
+            ),
         ):
 
             # Run the migrate command
@@ -195,7 +198,10 @@ def test_cli_config_loading(self, mock_api_key_check, facility_config_path):
                 "llama_prompt_ops.interfaces.cli.get_strategy", return_value=MagicMock()
             ),
             patch("llama_prompt_ops.interfaces.cli.load_config", return_value={}),
-            patch("llama_prompt_ops.interfaces.cli.validate_min_records_in_dataset", return_value=None),
+            patch(
+                "llama_prompt_ops.interfaces.cli.validate_min_records_in_dataset",
+                return_value=None,
+            ),
         ):
 
             # Run the migrate command with the real config
@@ -270,7 +276,10 @@ def test_end_to_end_cli_flow(self, mock_api_key_check, temp_config_file):
                     return_value=MagicMock(),
                 ),
                 patch("llama_prompt_ops.interfaces.cli.load_config", return_value={}),
-                patch("llama_prompt_ops.interfaces.cli.validate_min_records_in_dataset", return_value=None),
+                patch(
+                    "llama_prompt_ops.interfaces.cli.validate_min_records_in_dataset",
+                    return_value=None,
+                ),
             ):
 
                 # Run the migrate command with the actual file output