nbdev 3.0.13__tar.gz → 3.0.15__tar.gz

{nbdev-3.0.13/nbdev.egg-info → nbdev-3.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nbdev
-Version: 3.0.13
+Version: 3.0.15
 Summary: Create delightful software with Jupyter Notebooks
 Author-email: "Jeremy Howard and the fast.ai community" <j@fast.ai>
 License: Apache-2.0

{nbdev-3.0.13 → nbdev-3.0.15}/nbdev/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.0.13"
+__version__ = "3.0.15"
 
 from .doclinks import nbdev_export
 from .showdoc import show_doc

{nbdev-3.0.13 → nbdev-3.0.15}/nbdev/clean.py

@@ -58,7 +58,7 @@ def _clean_cell_output_id(lines):
     return _skip_or_sub(lines) if isinstance(lines,str) else [_skip_or_sub(o) for o in lines]
 
 # %% ../nbs/api/11_clean.ipynb #b4cde615
-def _clean_cell_output(cell, clean_ids):
+def _clean_cell_output(cell, clean_ids, allowed_out_meta_keys):
     "Remove `cell` output execution count and optionally ids from text reprs"
     outputs = cell.get('outputs', [])
     for o in outputs:
@@ -69,15 +69,15 @@ def _clean_cell_output(cell, clean_ids):
             if k.startswith('text') and clean_ids: data[k] = _clean_cell_output_id(data[k])
             if k.startswith('image') and "svg" not in k: data[k] = data[k].rstrip()
         if 'text' in o and clean_ids: o['text'] = _clean_cell_output_id(o['text'])
-#         o.get('metadata', {}).pop('tags', None)
+        if 'metadata' in o: o['metadata'] = {k:v for k,v in o['metadata'].items() if k in allowed_out_meta_keys}
 
 # %% ../nbs/api/11_clean.ipynb #2ba79c93
-def _clean_cell(cell, clear_all, allowed_metadata_keys, clean_ids):
+def _clean_cell(cell, clear_all, allowed_metadata_keys, clean_ids, allowed_out_meta_keys):
     "Clean `cell` by removing superfluous metadata or everything except the input if `clear_all`"
     if 'execution_count' in cell: cell['execution_count'] = None
     if 'outputs' in cell:
         if clear_all: cell['outputs'] = []
-        else:         _clean_cell_output(cell, clean_ids)
+        else:         _clean_cell_output(cell, clean_ids, allowed_out_meta_keys)
     if cell['source'] == ['']: cell['source'] = []
     cell['metadata'] = {} if clear_all else {
         k:v for k,v in cell['metadata'].items() if k in allowed_metadata_keys}
@@ -90,13 +90,16 @@ def clean_nb(
     allowed_metadata_keys:list=None, # Preserve the list of keys in the main notebook metadata
     allowed_cell_metadata_keys:list=None, # Preserve the list of keys in cell level metadata
     clean_ids=True, # Remove ids from plaintext reprs?
+    allowed_out_metadata_keys:list=None, # Preserve the list of keys in output metadata
 ):
     "Clean `nb` from superfluous metadata"
     metadata_keys = {"kernelspec", "jekyll", "jupytext", "doc", "widgets"}
     if allowed_metadata_keys: metadata_keys.update(allowed_metadata_keys)
     cell_metadata_keys = {"hide_input"}
     if allowed_cell_metadata_keys: cell_metadata_keys.update(allowed_cell_metadata_keys)
-    for c in nb['cells']: _clean_cell(c, clear_all, cell_metadata_keys, clean_ids)
+    out_meta_keys = set()
+    if allowed_out_metadata_keys: out_meta_keys.update(allowed_out_metadata_keys)
+    for c in nb['cells']: _clean_cell(c, clear_all, cell_metadata_keys, clean_ids, out_meta_keys)
     if nb.get('metadata', {}).get('kernelspec', {}).get('name', None):
         nb['metadata']['kernelspec']['display_name'] = nb["metadata"]["kernelspec"]["name"]
     nb['metadata'] = {k:v for k,v in nb['metadata'].items() if k in metadata_keys}
@@ -127,7 +130,8 @@ def _nbdev_clean(nb, path=None, clear_all=None):
     clear_all = clear_all or cfg.clear_all
     allowed_metadata_keys = cfg.get("allowed_metadata_keys") or []
     allowed_cell_metadata_keys = cfg.get("allowed_cell_metadata_keys") or []
-    clean_nb(nb, clear_all, allowed_metadata_keys, allowed_cell_metadata_keys, cfg.clean_ids)
+    allowed_out_metadata_keys = cfg.get("allowed_out_metadata_keys") or []
+    clean_nb(nb, clear_all, allowed_metadata_keys, allowed_cell_metadata_keys, cfg.clean_ids, allowed_out_metadata_keys)
     if path: nbdev_trust.__wrapped__(path)
 
 # %% ../nbs/api/11_clean.ipynb #6af3b9d4
@@ -209,7 +213,7 @@ def nbdev_install_hooks():
     os.chmod(fn, os.stat(fn).st_mode | stat.S_IEXEC)
 
     cmd = 'git config --local include.path ../.gitconfig'
-    (repo_path/'.gitconfig').write_text(f'''# Generated by nbdev_install_hooks
+    (repo_path/'.gitconfig').write_text(f'''# Generated by nbdev-install-hooks
 #
 # If you need to disable this instrumentation do:
 #   git config --local --unset include.path

{nbdev-3.0.13 → nbdev-3.0.15}/nbdev/maker.py

@@ -5,10 +5,6 @@
 # %% auto #0
 __all__ = ['find_var', 'read_var', 'update_var', 'ModuleMaker', 'decor_id', 'make_code_cells', 'relative_import', 'update_import']
 
-# %% ../nbs/api/02_maker.ipynb #aac32462
-#| export
-
-
 # %% ../nbs/api/02_maker.ipynb #ae4d461b
 from .config import *
 from .imports import *
@@ -23,6 +19,7 @@ import ast,contextlib
 from collections import defaultdict
 from pprint import pformat
 from textwrap import TextWrapper
+from itertools import dropwhile,takewhile
 
 # %% ../nbs/api/02_maker.ipynb #c2d38766
 def find_var(lines, varname):
@@ -178,12 +175,13 @@ def _import2relative(cells, lib_path=None):
 
 # %% ../nbs/api/02_maker.ipynb #5bff9d71
 def _retr_mdoc(cells):
-    "Search for md meta quote line, used to create module docstring"
+    "Search for md meta quote lines, used to create module docstring"
     md1 = first(o for o in cells if o.cell_type=='markdown' and o.source.startswith('# '))
     if not md1: return ''
-    summ = first(o for o in md1.source.splitlines() if o.startswith('> '))
-    if not summ: return ''
-    summ = summ.lstrip('> ').strip()
+    lines = dropwhile(lambda l: not l.startswith('> '), md1.source.splitlines())
+    lines = list(takewhile(lambda l: l.startswith('> '), lines))
+    if not lines: return ''
+    summ = '\n'.join(l.lstrip('> ').strip() for l in lines)
     return f'"""{summ}"""\n\n' if summ else ''
 
 # %% ../nbs/api/02_maker.ipynb #cdd205d6

{nbdev-3.0.13 → nbdev-3.0.15/nbdev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nbdev
-Version: 3.0.13
+Version: 3.0.15
 Summary: Create delightful software with Jupyter Notebooks
 Author-email: "Jeremy Howard and the fast.ai community" <j@fast.ai>
 License: Apache-2.0