rapidfireai 0.9.9__py3-none-any.whl → 0.9.11__py3-none-any.whl

rapidfireai/cli.py

@@ -10,6 +10,7 @@ import argparse
 import platform
 import shutil
 import re
+import site
 from pathlib import Path
 from .version import __version__
 
@@ -301,8 +302,6 @@ def install_packages():
     # Generate CUDA requirements file
     cuda_major = get_cuda_version()
     compute_capability = get_compute_capability()
-    print(f"CUDA major version: {cuda_major}")
-    print(f"Compute capability: {compute_capability}")
     if cuda_major == 12:
         print(f"\n🎯 Detected CUDA {cuda_major}.x")
         packages.append({"package": "vllm==0.10.1.1", "extra_args": ["--torch-backend=cu126"]})
@@ -313,7 +312,7 @@ def install_packages():
         print("\n⚠️  CUDA version not detected or unsupported.")
     if compute_capability == 7:
         print(f"\n🎯 Detected CUDA Compute Capability {compute_capability}.x")
-        packages.append({"package": "flash-attn==1.0.9", "extra_args": ["--no-build-isolation"]})
+        print("Skipping flash-attn installation")
     elif compute_capability == 8:
         print(f"\n🎯 Detected CUDA Compute Capability {compute_capability}.x")
         packages.append({"package": "flash-attn==2.8.3", "extra_args": ["--no-build-isolation"]})
@@ -333,19 +332,40 @@ def install_packages():
             print(f"   You may need to install {package} manually")
     return 0
 
+def copy_tutorial_notebooks():
+    """Copy the tutorial notebooks to the project."""
+    print("Getting tutorial notebooks...")
+    try:
+        tutorial_path = os.getenv("RF_TUTORIAL_PATH", os.path.join(".", "tutorial_notebooks"))
+        site_packages_path = site.getsitepackages()[0]
+        source_path =os.path.join(site_packages_path, "tutorial_notebooks")
+        print(f"Copying tutorial notebooks from {source_path} to {tutorial_path}...")
+        os.makedirs(tutorial_path, exist_ok=True)
+        shutil.copytree(source_path, tutorial_path, dirs_exist_ok=True)
+        print(f"✅ Successfully copied notebooks to {tutorial_path}")
+    except Exception as e:
+        print(f"❌ Failed to copy notebooks to {tutorial_path}")
+        print(f"   Error: {e}")
+        print("   You may need to copy notebooks manually")
+        return 1
+    return 0
+
+
 def run_init():
     """Run the init command to initialize the project."""
     print("🔧 Initializing RapidFire AI project...")
     print("-" * 30)
     print("Initializing project...")
     install_packages()
+    copy_tutorial_notebooks()
+
     return 0
 
 def main():
-    """Main entry point for the rapidfire-start command."""
+    """Main entry point for the rapidfireai command."""
     parser = argparse.ArgumentParser(
         description="RapidFire AI - Start/stop/manage services",
-        prog="rapidfire"
+        prog="rapidfireai"
     )
     
     parser.add_argument(

rapidfireai/experiment.py

@@ -5,7 +5,8 @@ This module contains the Experiment class which manages the entire experiment li
 import multiprocessing as mp
 import os
 import traceback
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 import pandas as pd
 from torch.utils.data import Dataset
@@ -17,6 +18,7 @@ from rapidfireai.utils.exceptions import ExperimentException
 from rapidfireai.utils.experiment_utils import ExperimentUtils
 from rapidfireai.utils.logging import RFLogger
 from rapidfireai.utils.mlflow_manager import MLflowManager
+from rapidfireai.version import __version__
 
 
 class Experiment:
@@ -61,6 +63,8 @@ class Experiment:
             self.logger = RFLogger().create_logger("experiment")
             for msg in log_messages:
                 self.logger.info(msg)
+            # Log the version of rapidfireai that is running
+            self.logger.info(f"Running RapidFire AI version {__version__}")
         except Exception as e:
             raise ExperimentException(f"Error creating logger: {e}, traceback: {traceback.format_exc()}") from e