onnx model info and doc fixes

doc/api/services/report_images.rst

@@ -3,4 +3,5 @@ sasctl.services.report_images
 
 .. automodule:: sasctl._services.report_images
     :members:
-    :undoc-members:
+    :undoc-members:
+    :show-inheritance:

doc/api/services/reports.rst

@@ -4,3 +4,4 @@ sasctl.services.reports
 .. automodule:: sasctl._services.reports
     :members:
     :undoc-members:
+    :show-inheritance:

doc/conf.py

@@ -109,7 +109,7 @@
     ('py:class','Response'),
     ('py:class','Request'),
     ('py:class','_io.BytesIO'),
-    ('py:class','sasctl.utils.pymas.ds2.Ds2Variable'), # not sure what is wrong 
+    ('py:class','sasctl.utils.pymas.ds2.Ds2Variable'), # not sure what is wrong
     ('py:class','sasctl._services.service.Service') # should the Service class be documented?
 ]
 

src/sasctl/_services/cas_management.py

@@ -31,7 +31,7 @@ def check_keys(valid_keys: list, input_keys: list, parameters: str):
 
     Raises
     ------
-    ValueError 
+    ValueError
         if input_keys are not valid
     """
     if not all(key in valid_keys for key in input_keys):

src/sasctl/_services/concepts.py

@@ -56,7 +56,7 @@ def assign_concepts(
         output_postfix : str, optional
             Text to be added to the end of all output table names.
         match_type : str, optional
-            Choose from ``{'all', 'longest', 'best'}``. 
+            Choose from ``{'all', 'longest', 'best'}``.
             Type of matches to return. Defaults to 'all'.
         enable_facts : bool, optional
             Whether to enable facts in the results.  Defaults to False.

src/sasctl/_services/data_sources.py

@@ -40,7 +40,7 @@ def get_provider(cls, provider, refresh=False):
             A dictionary containing the provider attributes or None.
 
         Notes
-        -------
+        -----
         If `provider` is a complete representation of the provider it will be
         returned unless `refresh` is set.  This prevents unnecessary REST calls
         when data is already available on the client.

src/sasctl/_services/folders.py

@@ -71,7 +71,7 @@ def get_folder(cls, folder, refresh=False):
         ----------
         folder : str or dict
             May be one of:
-            
+
             - folder name
             - folder ID
             - folder path

src/sasctl/_services/service.py

@@ -229,7 +229,7 @@ def get_item(cls, item, refresh=False):
                 A dictionary containing the {item} attributes or None.
 
             Notes
-            -------
+            -----
             If `item` is a complete representation of the {item} it will be
             returned unless `refresh` is set.  This prevents unnecessary REST
             calls when data is already available on the client.

src/sasctl/core.py

@@ -1254,7 +1254,7 @@ def _request_token_with_oauth(
         """Request a token from the SAS SASLogon service.
 
         Supports four different flows:
-        
+
         - authenticate with a username & password and receive a token
         - authenticate with a client id & secret and receive a token
         - provide an authorization code and receive a token
@@ -1531,6 +1531,13 @@ def __init__(self, obj, session=None, threads=4):
         # Store the current items to iterate over
         self._obj = obj
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self._pool is not None:
+            self._pool.shutdown(wait=False)
+
     def __next__(self):
         if self._pool is None:
             self._pool = concurrent.futures.ThreadPoolExecutor(
@@ -1778,7 +1785,7 @@ class VersionInfo:
         Release cadence for Viya 4.  Should be one of 'stable' or 'LTS'.
     release : str, optional
         Release number for Viya 4.  Two formats are currently possible:
-        
+
         - YYYY.R.U where R is the LTS release number in YYYY and U is the updates since R
         - YYYY.MM where MM is the month of the release.
 
@@ -2028,6 +2035,7 @@ def request(verb, path, session=None, format="auto", **kwargs):
 
     Returns
     -------
+    str, bytes, or requests.Response
 
     """
     session = session or current_session()
@@ -2111,6 +2119,7 @@ def request_link(obj, rel, **kwargs):
 
     Returns
     -------
+    RestObj
 
     """
     link = get_link(obj, rel)
@@ -2359,7 +2368,7 @@ def platform_version():
 
     Returns
     -------
-    str 
+    str
         SAS Viya version number '3.5' or '4.0'
 
     """

src/sasctl/pzmm/import_model.py

@@ -306,7 +306,7 @@ def import_model(
         **kwargs
             Other keyword arguments are passed to the following function:
             :meth:`.ScoreCode.write_score_code`
-            
+
 
         Returns
         -------

src/sasctl/pzmm/pickle_model.py

@@ -37,7 +37,7 @@ def pickle_trained_model(
         object.
 
         The following files are generated by this function:
-        
+
         * '\*.pickle'
             Binary pickle file containing a trained model.
         * '\*.mojo'

src/sasctl/pzmm/write_json_files.py

@@ -5,7 +5,8 @@
 import ast
 import importlib
 import json
-# import math #not used 
+
+# import math #not used
 import pickle
 import pickletools
 import sys

src/sasctl/pzmm/zip_model.py

@@ -71,7 +71,7 @@ def zip_files(
         """
 
         if isinstance(model_files, dict):
-            buffer = io.BytesIO()
+            buffer = BytesIO()
 
             with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED, False) as archive:
                 for file_name, data in model_files.items():