pythoncharmers-meta 0.7.5__tar.gz → 0.7.7__tar.gz

{pythoncharmers_meta-0.7.5 → pythoncharmers_meta-0.7.7}/.gitignore

@@ -160,3 +160,12 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+# macOS rubbish:
+.DS_Store
+Icon?
+# Emacs, VIM, and VS Code rubbish:
+*~
+*.swp
+*.swo
+.vscode/

{pythoncharmers_meta-0.7.5 → pythoncharmers_meta-0.7.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pythoncharmers-meta
-Version: 0.7.5
+Version: 0.7.7
 Summary: Meta package with dependencies for Python Charmers training courses
 License-File: LICENSE
 Requires-Python: >=3.10
@@ -9,6 +9,7 @@ Requires-Dist: diskcache>=5.6.3
 Requires-Dist: humanize>=4.10.0
 Requires-Dist: hvplot>=0.10.0
 Requires-Dist: ipython>=8.12.3
+Requires-Dist: ipywidgets>=8.1.7
 Requires-Dist: isort>=5.13.2
 Requires-Dist: keyring>=25.4.1
 Requires-Dist: keyrings-cryptfile>=1.3.9
@@ -88,7 +89,9 @@ Requires-Dist: pyodbc>=5.2.0; extra == 'db'
 Requires-Dist: redis>=5.2.1; extra == 'db'
 Requires-Dist: sqlalchemy>=2.0.38; extra == 'db'
 Provides-Extra: dl
+Requires-Dist: albumentations>=2.0.8; extra == 'dl'
 Requires-Dist: dataset>=1.5.2; extra == 'dl'
+Requires-Dist: datasets>=2.2.1; extra == 'dl'
 Requires-Dist: diskcache>=5.6.3; extra == 'dl'
 Requires-Dist: eli5>=0.15.0; extra == 'dl'
 Requires-Dist: evaluate>=0.4.3; extra == 'dl'
@@ -100,7 +103,7 @@ Requires-Dist: scikit-learn>=1.6.1; extra == 'dl'
 Requires-Dist: skorch>=1.0.0; extra == 'dl'
 Requires-Dist: tensorboard>=2.18.0; extra == 'dl'
 Requires-Dist: textdistance[levenshtein]>=4.6.3; extra == 'dl'
-Requires-Dist: transformers>=4.49.0; extra == 'dl'
+Requires-Dist: transformers[torch]>=4.49.0; extra == 'dl'
 Requires-Dist: xgboost>=2.1.1; extra == 'dl'
 Requires-Dist: yellowbrick>=1.5; extra == 'dl'
 Provides-Extra: docs
@@ -140,6 +143,7 @@ Requires-Dist: xarray>=2024.9.0; extra == 'gis'
 Requires-Dist: zarr>=2.18.3; extra == 'gis'
 Provides-Extra: ml
 Requires-Dist: eli5>=0.15.0; extra == 'ml'
+Requires-Dist: glum>=3.1.2; extra == 'ml'
 Requires-Dist: joblib>=1.4.2; extra == 'ml'
 Requires-Dist: pyod>=2.0.5; extra == 'ml'
 Requires-Dist: scikit-learn>=1.6.1; extra == 'ml'

{pythoncharmers_meta-0.7.5 → pythoncharmers_meta-0.7.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pythoncharmers-meta"
-version = "0.7.5"
+version = "0.7.7"
 description = "Meta package with dependencies for Python Charmers training courses"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -10,6 +10,7 @@ dependencies = [
     "humanize>=4.10.0",
     "hvplot>=0.10.0",
     "ipython>=8.12.3",
+    "ipywidgets>=8.1.7",
     "isort>=5.13.2",
     "keyring>=25.4.1",
     "keyrings-cryptfile>=1.3.9",
@@ -77,6 +78,7 @@ gis = [
 ]
 ml = [
     "eli5>=0.15.0",
+    "glum>=3.1.2",
     "joblib>=1.4.2",
     "pyod>=2.0.5",
     "scikit-learn>=1.6.1",
@@ -194,7 +196,9 @@ net = [
     "uvicorn>=0.34.0",
 ]
 dl = [
+    "albumentations>=2.0.8",
     "dataset>=1.5.2",
+    "datasets>=2.2.1",
     "diskcache>=5.6.3",
     "eli5>=0.15.0",
     "evaluate>=0.4.3",
@@ -206,7 +210,7 @@ dl = [
     "skorch>=1.0.0",
     "tensorboard>=2.18.0",
     "textdistance[Levenshtein]>=4.6.3",
-    "transformers>=4.49.0",
+    "transformers[torch]>=4.49.0",
     "xgboost>=2.1.1",
     "yellowbrick>=1.5",
 ]

{pythoncharmers_meta-0.7.5 → pythoncharmers_meta-0.7.7}/src/pythoncharmers_meta/__init__.py

@@ -37,6 +37,7 @@ path and/or file that %code and %md query. To get help on these, run:
 # Import version from package metadata (single source of truth)
 try:
     from importlib.metadata import version, PackageNotFoundError
+
     try:
         __version__ = version("pythoncharmers-meta")
     except PackageNotFoundError:
@@ -47,6 +48,7 @@ except ImportError:
     # Python < 3.8 fallback
     try:
         import pkg_resources
+
         __version__ = pkg_resources.get_distribution("pythoncharmers-meta").version
     except Exception:
         __version__ = "unknown"
@@ -58,6 +60,7 @@ from .nb_magic import NotebookMagic
 
 try:
     from .ai_magic import AIMagic
+
     __all__.append("AIMagic")
 except Exception as e:
     print(e)

pythoncharmers_meta-0.7.7/src/pythoncharmers_meta/ai_magic.py

@@ -0,0 +1,61 @@
+from IPython.core.magic import Magics, magics_class, line_cell_magic
+from IPython.core.magic_arguments import argument, magic_arguments, parse_argstring
+import llm
+
+
+@magics_class
+class AIMagic(Magics):
+    def __init__(self, shell, default_model_name="gpt-4o-mini"):
+        # You must call the parent constructor
+        super().__init__(shell)
+        self.default_model = llm.get_model(default_model_name)
+
+    @magic_arguments()
+    @argument("-m", "--model", type=str, help="Model name to use")
+    @argument("-t", "--temperature", type=float, help="Sampling temperature")
+    @argument(
+        "prompt",
+        nargs="*",
+        help="Prompt text (required for line magic, optional for cell magic)",
+    )
+    @line_cell_magic
+    def ai(self, line, cell=None):
+        # Parse arguments
+        args = parse_argstring(self.ai, line)
+        options = vars(args)
+
+        # Combine cell content and line prompt
+        line_prompt = " ".join(args.prompt)
+
+        # Check if we have any prompt content
+        if not line_prompt and cell is None:
+            print("Error: No prompt provided.")
+            print("\nUsage:")
+            print("  Line magic:  %ai <prompt>")
+            print("  Cell magic:  %%ai [prompt]")
+            print("               <cell content as prompt>")
+            print("\nOptions:")
+            print("  -m, --model <name>        Model name to use")
+            print("  -t, --temperature <float> Sampling temperature")
+            print("\nExample:")
+            print("  %ai What is Python?")
+            print("  %ai -m gpt-4 Explain recursion")
+            return
+
+        prompt = line_prompt + ("" if cell is None else "\n\n" + cell)
+
+        if "model" in options:
+            model = llm.get_model(options["model"])
+        else:
+            model = self.default_model
+
+        # Prepare options to pass to llm
+        completion_args = {
+            k: v
+            for k, v in options.items()
+            if k not in {"prompt", "model"} and v is not None
+        }
+
+        # Call llm and return response
+        response = model.prompt(prompt, **completion_args)
+        print(response.text())

{pythoncharmers_meta-0.7.5 → pythoncharmers_meta-0.7.7}/src/pythoncharmers_meta/nb_magic.py

@@ -531,9 +531,6 @@ class NotebookMagic(Magics):
 
         Note: After running, convert the cell to Markdown with Esc M in Jupyter.
         """
-        # Store original arg_s for better error messages
-        original_arg_s = arg_s
-
         # Check for common mistakes (but not --list itself!)
         if "--list" not in arg_s and (
             "--lsit" in arg_s
@@ -542,9 +539,9 @@ class NotebookMagic(Magics):
             or "--lits" in arg_s
         ):
             raise UsageError(
-                f"Invalid option. Did you mean '--list'?\n"
-                f"Usage: %md --list  or  %md m1-m3  or  %md -1 2-\n"
-                f"See %md? for full documentation."
+                "Invalid option. Did you mean '--list'?\n"
+                "Usage: %md --list  or  %md m1-m3  or  %md -1 2-\n"
+                "See %md? for full documentation."
             )
 
         # Check if user is trying to use old %mdat syntax
@@ -736,7 +733,7 @@ class NotebookMagic(Magics):
                     if content_hash not in seen_content:
                         seen_content.add(content_hash)
                         contents.append(content)
-            except (ValueError, IndexError) as e:
+            except (ValueError, IndexError):
                 # Check if it's a markdown index that's out of range
                 if "m" in arg.lower():
                     all_markdown = get_markdown_cells(nb)
@@ -764,8 +761,8 @@ class NotebookMagic(Magics):
                 )
             else:
                 raise UsageError(
-                    f"No markdown cells found for the specified ranges.\n"
-                    f"Use '%md --list' to see available markdown cells."
+                    "No markdown cells found for the specified ranges.\n"
+                    "Use '%md --list' to see available markdown cells."
                 )
             return