Merge pull request #8 from jlwalke2/docstring-changes

Docstring changes

Author: Jon Walker

README.md

@@ -5,13 +5,15 @@
   <p>A user-friendly REST client for SAS Viya.</p>
 
   <a href="https://www.sas.com/en_us/software/viya.html">
-    <img src="https://img.shields.io/badge/SAS%20Viya-3.4-blue.svg?&colorA=0b5788&style=for-the-badge&logoWidth=30&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAADMAAABLCAYAAADd0L+GAAAJ+ElEQVR42t1beVCV1xU/z5n+VW2S2rSxjdla0zrWRGubSa21ndpO28TUJm1GsWpiVRKsCkZrFaPGojRsj4CyPUCQHQUBMbJvKoqRRaMiahaFJDqKsj3e4y1gzw/mo1CW797vvTycvPEMI/O9+93fvWf9nQN9feG7XxlxyiLjWSYuCaTvLg+mn6yPpsVBh2hHSjnFFNZS6rHzdOXz5imQYxevU3LFeTLw771iC+gvfgfpsZUh9Mjrenpgsf/YgnmQN/CzjTHkHp5LSeXnqc1o9rl37163jPDHDKC+Gcdpwe50euqNPa4H84vNcRR+9AzdvNvxAjaEjTkiPT093VabrR63t55vbfKKEHkwsur083/to4i8arLb7XexAaHNyt+Wrb2zS//WvkJ6YlUojXc2mG8vC6KVYbnU0m7ykt6gdlDW7KoG+sPOZBq/yElgfrg6nAz5NWSz25vwElcK65/NYrVVro48St9aGugYmJnrDVRx4Rph4bEUu73bGJFfTU+4h2oD86xnFBXUfkQ4nbEGo3i+fcV19NDf/OXAzFgfRU3NrXOcaeRYix1He4fZYoAXvNR4a2LJuU9IkeP1jfTpzZbpcPHwbDjE4ZzD/tJz9NiK98TAwINkn24gZ55o4+3Wmb4ZJ2jl3lyaty2Rpq+LpEf/PnhDD7OTmeIRRnO2xNOr/hn0dnIp5Zy+TBab7fSAQ8WBtLyTWkEPuPmPDgZG5n+okrABJ9zCjcyT9TR/VyqCoXSUn7DIrzermLomgjbGFdGVz5qn4GYVQC/tThsdDFIMm83e5CiQ989cpZf/c0A5PafIo6xa8GqNt1pmQQUvXLs5aeo/wocH89CSAIIeOwICsSGqoIa+L5SWyAvizawN0RRXUofAfWt7Snn/gQ16yCumAOpl0QrGbLEeXRaSzerhmix5A2cIVQ1NE57/Z+xgMPDfhXUfk1bvBftYFXaEvuHm57KU/5usSZsTSmBPg8H8tc9WrmtRLURo9/AjbOAKENcJSo8NcYU4xD4w8DJB2adIa1L4dnIZB7KAMSvKHnktiN16YB+Y7/B/Khsav6blVo5dvIbi6v6pNJ90D9Vk+FCv32xLFH0ZYphSWX55YOZ6x5OWW0koO4eNCZUPS4Kz6GBlPeVzrnfo1CVCrQJgzgaD4CYNBs5iUWCmQPkQ1guCs147f68Hgg9rQk/J2U9QUToVDMgFaTCtHabNj68KUfE0AZRQ9iEBwEgSU1SLG3IaGHZtRdJgkHOpLf4n33R297bm0cBwfLJuSy5DzBg7NfNOKlVdHO4exoVNqwCyvRn5vlPAICWXBrMmKk91ceRo2KyIdFks5b/bkeQoGNQvIdKueXlojurim+KLCVFVBAw+TZwNz/Xe7xgYuFdUfs5Ws5lvRVOr0bQJmxUV8A0oDjWDgfGhFJUBE5lfLZSuLwzIRKpuFgUDG4stqsUBaycBl4XkEBgQUTAogxHRBShclBYAZBIFhBikzz6FfEsbGHDGX9xp/61w7WK1Fs/bLpLKIPfT91K5MuoG8EuDs7WBGc8SfLiK+FBsouQcnn9QsK5HZp77wWU4BGFAHKNa5/ukjlQj6ZSfigx64KcbYqRqmjttnSuUKk9EZjChCGIcnkvYw91umTV7c9zwYAYLDTFYQ0ENXiZMnRoKa3BywmwLaKQOk1kvYz8nLjWOe3xliG44EKOwM7idaLrb1ukhU5yhuSRT97+0K42Y5PtCxoa4aaVjdkanYjODEcIGkCvxJjtFSwF0BuZJ1DWgV7cklMDDWUTBIOv2TizBd0cFM+7/r47rD1368Ys6mdqmudW4DLcq3nXzI5TbMg4Bz3pGFwjdjCL96oaGj0wgPXz6slQbD4ERtY6Mulks1kp07aSIc9jAa8yBdVltFaIOAfkdksvJQ0ntEb3RtLWRuqPVV6lbwsPh+ac99oqDUezHMyZfinfGs2i2qsQFGiizubXY0tHpJaNuO9NAnPuJg1GqRUNBLdy1DCHY7KaU1IKyRJ8lZT/sDT+duiZ80C0LvWgyl7Up3M8HjywKqMNkiViwOw2xRdDDBVBA1kkpQLHFtTrOLPptXTx6e0XRifrGcdioeDLaMnOWhId7bmMs3e3o9BAFY+6yFM7dEq/T1Dr/JUdvU5c1U8Zl59V+xB4uVDhD6LudHuFyISjnVH/skW4nINoz258r0/6OLzkrysCg/Silas1tRrcfr41UwMgz71sTS4UzBAiexSyNyHACQoLR3GWQ8Wwv+6Y7NG6CckG6VYhOg8BwApyNVCBFcuwQGPDTWVUNUm11pP9TlGA3ivgcOAYwMqr2isNTTc+yhytnAkKGaAdHp7IuSEnZqvSzJ1eFOj5v9vymWEIJLQIG4ypwIGprbksplwVzA/maUwbnPJiNxBCCWpbQburSi7RAwD9LgIETaH/VL0MIjAgDg76iqodLLP+QJqpzykystM2RBGNaHJSlCkaqkRRbVDei/dxu7ViIqQy1dbg8JnDPkmBsChjdENEICOMj+pwqjhOWeAzXQdBOT+aRx2fWRQmp7NakUpmgqVShtj/+O4VIcPNSJfGvtu6nFXsOQzD4JqRakKdXh0mxN4qg/P4Rf/e+GeNF5F8XnS+tYhD0gJTW+X0hzzGjipJYFggEjS/cPhbqLXN/8ObeMQPyPba1DN6QFiCQN8KPoHPwvzmALYklAOVyIHhneF61YvTSYjSZDTO8DBjl6gMDfcPIBobbBLljp8Unbo0AiF0LENQzIFCUbsEAUiGOPrjy+cTA7JPw9SrpuuNZA+r38LwzWm9EoZ3OvOiTOpTQmMC3AyaTfbYlr+YqvcB++8uYUMKav9+ZxBO51xV6SbPgVgcyNEOC3q3Wjj/jQVOXJXf3weMg9ZxnH7z+Lk7vjWazSvElRgZOWxsxOtUEzhidXwQufBCQ9hWfJRRWz3hGwQVKzVii7sGaPCCKdkmnsq4jQEC6c/Y9xBSGo3ww1zKkDwkj/fhG8zQki+8wAefGi/16awJNZ4ADBR24+T5pva0/PVejmJWxWK0XVFRKim/ekVKGeRwxRhMDaT7pFQQAIy2IG0PkxUYHitVqu4obwHfVAcgDiSuuG3GMflS36Zd5ov+GxlpwOGzwHGCDtY3PT2KW3puZGPRGFD13teCDG4YzUqOr1HqFymwNCqbZjsQErUHxTrvx9aXBWSKduZHqmcENKPZKOm7e6qILa3WuAoT3YIQfHQIFiBAYUYHhvcij8Pk8Mgzjd7LqKaHACk57IXcRJi1X7EM7GFKThxnUK+8eoDimXaEGzgACL4i/FMR4PGzV5X8NiGwb3Nny0MMUX3qWkMHa2etARRThfwOke6DY2ZXXZlVdIs/ofJDyyk1oFqcnkE+57yHU4/jTkh2p5Uhf+mU7Bzv8foFvOkpkgd6NPJivjPwX66dH9VYtHvAAAAAASUVORK5CYII="
+    <img src="https://img.shields.io/badge/SAS%20Viya-3.4-blue.svg?&colorA=0b5788&logoWidth=30&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAADMAAABLCAYAAADd0L+GAAAJ+ElEQVR42t1beVCV1xU/z5n+VW2S2rSxjdla0zrWRGubSa21ndpO28TUJm1GsWpiVRKsCkZrFaPGojRsj4CyPUCQHQUBMbJvKoqRRaMiahaFJDqKsj3e4y1gzw/mo1CW797vvTycvPEMI/O9+93fvWf9nQN9feG7XxlxyiLjWSYuCaTvLg+mn6yPpsVBh2hHSjnFFNZS6rHzdOXz5imQYxevU3LFeTLw771iC+gvfgfpsZUh9Mjrenpgsf/YgnmQN/CzjTHkHp5LSeXnqc1o9rl37163jPDHDKC+Gcdpwe50euqNPa4H84vNcRR+9AzdvNvxAjaEjTkiPT093VabrR63t55vbfKKEHkwsur083/to4i8arLb7XexAaHNyt+Wrb2zS//WvkJ6YlUojXc2mG8vC6KVYbnU0m7ykt6gdlDW7KoG+sPOZBq/yElgfrg6nAz5NWSz25vwElcK65/NYrVVro48St9aGugYmJnrDVRx4Rph4bEUu73bGJFfTU+4h2oD86xnFBXUfkQ4nbEGo3i+fcV19NDf/OXAzFgfRU3NrXOcaeRYix1He4fZYoAXvNR4a2LJuU9IkeP1jfTpzZbpcPHwbDjE4ZzD/tJz9NiK98TAwINkn24gZ55o4+3Wmb4ZJ2jl3lyaty2Rpq+LpEf/PnhDD7OTmeIRRnO2xNOr/hn0dnIp5Zy+TBab7fSAQ8WBtLyTWkEPuPmPDgZG5n+okrABJ9zCjcyT9TR/VyqCoXSUn7DIrzermLomgjbGFdGVz5qn4GYVQC/tThsdDFIMm83e5CiQ989cpZf/c0A5PafIo6xa8GqNt1pmQQUvXLs5aeo/wocH89CSAIIeOwICsSGqoIa+L5SWyAvizawN0RRXUofAfWt7Snn/gQ16yCumAOpl0QrGbLEeXRaSzerhmix5A2cIVQ1NE57/Z+xgMPDfhXUfk1bvBftYFXaEvuHm57KU/5usSZsTSmBPg8H8tc9WrmtRLURo9/AjbOAKENcJSo8NcYU4xD4w8DJB2adIa1L4dnIZB7KAMSvKHnktiN16YB+Y7/B/Khsav6blVo5dvIbi6v6pNJ90D9Vk+FCv32xLFH0ZYphSWX55YOZ6x5OWW0koO4eNCZUPS4Kz6GBlPeVzrnfo1CVCrQJgzgaD4CYNBs5iUWCmQPkQ1guCs147f68Hgg9rQk/J2U9QUToVDMgFaTCtHabNj68KUfE0AZRQ9iEBwEgSU1SLG3IaGHZtRdJgkHOpLf4n33R297bm0cBwfLJuSy5DzBg7NfNOKlVdHO4exoVNqwCyvRn5vlPAICWXBrMmKk91ceRo2KyIdFks5b/bkeQoGNQvIdKueXlojurim+KLCVFVBAw+TZwNz/Xe7xgYuFdUfs5Ws5lvRVOr0bQJmxUV8A0oDjWDgfGhFJUBE5lfLZSuLwzIRKpuFgUDG4stqsUBaycBl4XkEBgQUTAogxHRBShclBYAZBIFhBikzz6FfEsbGHDGX9xp/61w7WK1Fs/bLpLKIPfT91K5MuoG8EuDs7WBGc8SfLiK+FBsouQcnn9QsK5HZp77wWU4BGFAHKNa5/ukjlQj6ZSfigx64KcbYqRqmjttnSuUKk9EZjChCGIcnkvYw91umTV7c9zwYAYLDTFYQ0ENXiZMnRoKa3BywmwLaKQOk1kvYz8nLjWOe3xliG44EKOwM7idaLrb1ukhU5yhuSRT97+0K42Y5PtCxoa4aaVjdkanYjODEcIGkCvxJjtFSwF0BuZJ1DWgV7cklMDDWUTBIOv2TizBd0cFM+7/r47rD1368Ys6mdqmudW4DLcq3nXzI5TbMg4Bz3pGFwjdjCL96oaGj0wgPXz6slQbD4ERtY6Mulks1kp07aSIc9jAa8yBdVltFaIOAfkdksvJQ0ntEb3RtLWRuqPVV6lbwsPh+ac99oqDUezHMyZfinfGs2i2qsQFGiizubXY0tHpJaNuO9NAnPuJg1GqRUNBLdy1DCHY7KaU1IKyRJ8lZT/sDT+duiZ80C0LvWgyl7Up3M8HjywKqMNkiViwOw2xRdDDBVBA1kkpQLHFtTrOLPptXTx6e0XRifrGcdioeDLaMnOWhId7bmMs3e3o9BAFY+6yFM7dEq/T1Dr/JUdvU5c1U8Zl59V+xB4uVDhD6LudHuFyISjnVH/skW4nINoz258r0/6OLzkrysCg/Silas1tRrcfr41UwMgz71sTS4UzBAiexSyNyHACQoLR3GWQ8Wwv+6Y7NG6CckG6VYhOg8BwApyNVCBFcuwQGPDTWVUNUm11pP9TlGA3ivgcOAYwMqr2isNTTc+yhytnAkKGaAdHp7IuSEnZqvSzJ1eFOj5v9vymWEIJLQIG4ypwIGprbksplwVzA/maUwbnPJiNxBCCWpbQburSi7RAwD9LgIETaH/VL0MIjAgDg76iqodLLP+QJqpzykystM2RBGNaHJSlCkaqkRRbVDei/dxu7ViIqQy1dbg8JnDPkmBsChjdENEICOMj+pwqjhOWeAzXQdBOT+aRx2fWRQmp7NakUpmgqVShtj/+O4VIcPNSJfGvtu6nFXsOQzD4JqRakKdXh0mxN4qg/P4Rf/e+GeNF5F8XnS+tYhD0gJTW+X0hzzGjipJYFggEjS/cPhbqLXN/8ObeMQPyPba1DN6QFiCQN8KPoHPwvzmALYklAOVyIHhneF61YvTSYjSZDTO8DBjl6gMDfcPIBobbBLljp8Unbo0AiF0LENQzIFCUbsEAUiGOPrjy+cTA7JPw9SrpuuNZA+r38LwzWm9EoZ3OvOiTOpTQmMC3AyaTfbYlr+YqvcB++8uYUMKav9+ZxBO51xV6SbPgVgcyNEOC3q3Wjj/jQVOXJXf3weMg9ZxnH7z+Lk7vjWazSvElRgZOWxsxOtUEzhidXwQufBCQ9hWfJRRWz3hGwQVKzVii7sGaPCCKdkmnsq4jQEC6c/Y9xBSGo3ww1zKkDwkj/fhG8zQki+8wAefGi/16awJNZ4ADBR24+T5pva0/PVejmJWxWK0XVFRKim/ekVKGeRwxRhMDaT7pFQQAIy2IG0PkxUYHitVqu4obwHfVAcgDiSuuG3GMflS36Zd5ov+GxlpwOGzwHGCDtY3PT2KW3puZGPRGFD13teCDG4YzUqOr1HqFymwNCqbZjsQErUHxTrvx9aXBWSKduZHqmcENKPZKOm7e6qILa3WuAoT3YIQfHQIFiBAYUYHhvcij8Pk8Mgzjd7LqKaHACk57IXcRJi1X7EM7GFKThxnUK+8eoDimXaEGzgACL4i/FMR4PGzV5X8NiGwb3Nny0MMUX3qWkMHa2etARRThfwOke6DY2ZXXZlVdIs/ofJDyyk1oFqcnkE+57yHU4/jTkh2p5Uhf+mU7Bzv8foFvOkpkgd6NPJivjPwX66dH9VYtHvAAAAAASUVORK5CYII="
         alt="SAS Viya Version"/>
   </a>
         
   <a href="https://www.python.org/">
-    <img src="https://img.shields.io/badge/Python-3+-blue.svg?&style=for-the-badge&colorA=254f73" alt="Python Version">
+    <img src="https://img.shields.io/badge/Python-3+-blue.svg" alt="Python Version">
   </a>
+
+   <img src="https://travis-ci.com/jlwalke2/python-sasctl.svg?branch=master">
 
 </div>
 

src/sasctl/__init__.py

@@ -6,7 +6,7 @@
 
 __version__ = '0.9.6'
 __author__ = 'SAS'
-__credits__ = ['Lucas De Paula, Peter Tobac, Jon Walker']
+__credits__ = ['Yi Jian Ching, Lucas De Paula, Peter Tobac, Jon Walker']
 __license__ = 'Apache 2.0'
 __copyright__ = 'Copyright © 2019, SAS Institute Inc., ' \
                 'Cary, NC, USA.  All Rights Reserved.'

src/sasctl/services/model_management.py

@@ -50,7 +50,49 @@ def publish_model(model, destination, name=None, force=False):
     return r
 
 
-def create_performance_definition(model, library_name, table_name, name=None, description=None, cas_server=None):
+def create_performance_definition(model, library_name, table_name, name=None, description=None, outputLibrary=None, cas_server=None):
+    """Create the performance task definition in the model project to monitor model performance.
+
+    Parameters
+    ----------
+    model : str or dict
+        The name or id of the model, or a dictionary representation of the model.
+    library_name : str
+        The library containing the input data, default is 'Public'.
+    table_name : str
+        The name used for the performance data.
+    name : str
+        The name of the performance task, default is 'Performance'.
+    description : str
+        The description of the performance task, default is 'Performance monitoring for model' + model.name.
+    cas_server : str
+        The CAS Server for the monitoring task, default is 'cas-shared-default'.
+    championMonitored : bool
+        Indicates to monitor the project champion model.
+    challengerMonitored : bool
+        Indicates to monitor challenger models.
+    includeAllData : bool
+        Indicates whether to run a performance job against all the data tables in a library.
+    scoreExecutionRequired : bool
+        Indicates whether the scoring task execution is required. This should be set 'False' if you have provided the scores and 'True' if not.
+    maxBins : int
+        The maximum bins number, default is 10.
+    resultLibrary : str
+        The performance output table library, default is 'ModelPerformanceData'.
+    traceOn : bool
+        Indicates whether to turn on tracing.
+    performanceResultSaved : bool
+        Indicates whether the performance results are saved.
+    loadPerformanceResult : bool
+        Indicates to load performance result data.
+
+
+    Returns
+    -------
+    str
+        Performance task definition schema in JSON format.
+
+    """
     from .model_repository import get_model, get_project
 
     model = get_model(model)
@@ -62,13 +104,21 @@ def create_performance_definition(model, library_name, table_name, name=None, de
             raise ValueError("Project %s must have the '%s' property set." % (project.name, required))
 
     request = {'projectId': project.id,
-               'modelIds': [model.id],
                'name': name or model.name + ' Performance',
+               'modelIds': [model.id],
+               'championMonitored': False,
+               'challengerMonitored': False,
+               'includeAllData': False,
+               'scoreExecutionRequired': False,
+               'maxBins': 10,
+               'resultLibrary': outputLibrary or 'ModelPerformanceData',
+               'traceOn': False,
+               'performanceResultSaved': True,
+               'dataLibrary': library_name or 'Public',
+               'loadPerformanceResult': False,
                'description': description or 'Performance definition for model ' + model.name,
                'casServerId': cas_server or 'cas-shared-default',
-               'resultLibrary': 'ModelPerformanceData',
-               'dataLibrary': library_name,
-               'dataTable': table_name
+               'dataPrefix': table_name
                }
 
     # If model doesn't specify input/output variables, try to pull from project definition
@@ -79,4 +129,4 @@ def create_performance_definition(model, library_name, table_name, name=None, de
         request['inputVariables'] = [v.get('name') for v in project.get('variables', []) if v.get('role') == 'input']
         request['outputVariables'] = [v.get('name') for v in project.get('variables', []) if v.get('role') == 'output']
 
-    return post(SERVICE_ROOT + '/performanceTasks', json=request, headers={'Content-Type': 'application/vnd.sas.models.performance.task+json'})
+    return post(SERVICE_ROOT + '/performanceTasks', json=request, headers={'Content-Type': 'application/vnd.sas.models.performance.task+json'})

src/sasctl/services/model_repository.py

@@ -4,6 +4,7 @@
 # Copyright © 2019, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+import sys
 from sasctl.core import is_uuid, get, post, get_link, _build_crud_funcs, current_session
 
 
@@ -52,6 +53,18 @@ def get_model_link(model, rel, refresh=False):
 
 
 def get_astore(model):
+    """Get the ASTORE for a model registered int he model repository.
+
+    Parameters
+    ----------
+    model :  str or dict
+        The name or id of the model, or a dictionary representation of the model.
+
+    Returns
+    ----------
+    binary?
+
+    """
     # TODO: Download binary object?
 
     link = get_model_link(model, 'analyticStore', refresh=True)
@@ -61,7 +74,7 @@ def get_astore(model):
 
 
 def get_score_code(model):
-    """The score code for a model registered in the model repository.
+    """Get the score code for a model registered in the model repository.
 
     Parameters
     ----------
@@ -83,7 +96,7 @@ def get_score_code(model):
 
 
 def get_model_contents(model):
-    """The additional files and data associated with the model.
+    """Retrieve the additional files and data associated with the model.
 
     Parameters
     ----------
@@ -106,47 +119,61 @@ def get_model_contents(model):
 def create_model(model, project, description=None, modeler=None, function=None, algorithm=None, tool=None,
                  is_champion=False,
                  properties={}, **kwargs):
-    """
+    """Creates a model into a project or folder.
 
     Parameters
     ----------
-    model
-    project
+    model : str or dict
+        The name or id of the model, or a dictionary representation of the model.
+    project : str or dict
+        The name or id of the model project, or a dictionary representation of the model project.
     description : str, optional
+        The description of the model.
     modeler : str, optional
         Name of the user that created the model.  Current user name will be used if unspecified.
+    function : str, optional
+        The function of the model, valid values include: analytical, classification, cluster, forecasting, prediction, Text analytics, transformation.
+    algorithm : str, optional
+        The name of the model algorithm.
+    tool : str, optional
+        The name of the model tool, can be 'Python 2' or 'Python 3'.
+    scoreCodeType : str, optional
+        The score code type for the model.
+    trainTable : str, optional
+        The train data table.
+    classificationEventProbabilityVariableName : str, optional
+        The name of the event probability variable.
+    classificationTargetEventValue : str, optional
+        The target event value.
+    champion : bool, optional
+        Indicates whether the project has champion model or not.
+    role : str, optional
+        The role of the model, valid values include: plain, champion, challenger.
+    location : str, optional,
+        The location of this model.
+    targetVariable : str, optional
+        The name of the target variable.
+    suggestedChampion : bool
+        Indicates the model was suggested as the champion at import time.
+    retrainable : bool
+        Indicates whether the model can be retrained or not.
+    immutable : bool
+        Indicates whether the model can be changed or not.
+    modelVersionName : str, optional
+        The display name for the model version.
+    properties : array_like, optional (custom properties)
+        Custom model properties that can be set: name, value, type
+
+    inputVariables : array_like, optional
+        Model input variables. By default, these are the same as the model project.
+    outputVariables : array_like, optional
+        Model output variables. By default, these are the same as the model project.
 
-    function
-    algorithm
-    tool
-    modeler
-    scoreCodeType
-    trainTable
-    classificationEventProbabilityVariableName
-    classificationTargetEventValue
-    champion (T/F)
-    role
-    location
-    targetVariable
-    projectId, projectName, projectVersionId, projectVersionName???
-    suggestedChampion (T/F)
-    retrainable
-    immutable
-    modelVersionName
-    properties  (custom properties)
-        name
-        value
-        type
-    inputVariables
-        -
-    outputVariables
-        -
-
-    properties
-    kwargs
 
     Returns
     -------
+    str
+        The model schema returned in JSON format.
 
     """
 
@@ -177,6 +204,27 @@ def create_model(model, project, description=None, modeler=None, function=None,
 
 
 def add_model_content(model, file, name=None, role=None):
+    """Add additional files to the model.
+
+    Parameters
+    ----------
+    model : str or dict
+        The name or id of the model, or a dictionary representation of the model.
+    file : str or bytes
+        A file related to the model, such as the model code.
+    name : str
+        Name of the file related to the model.
+    role : str
+        Role of the model file, such as 'Python pickle'.
+
+
+    Returns
+    -------
+    str
+        The model content schema.
+
+    """
+
     if is_uuid(model):
         id = model
     elif isinstance(model, dict) and 'id' in model:
@@ -211,6 +259,22 @@ def default_repository():
 
 
 def create_project(project, repository, **kwargs):
+    """Create a model project in the given model repository.
+
+    Parameters
+    ----------
+    project : str or dict
+        The name or id of the model project, or a dictionary representation of the project.
+    repository : str or dict
+        The name or id of the model repository, or a dictionary representation of the repository.
+
+    Returns
+    -------
+    RestObj
+
+    """
+
+
     if isinstance(project, str):
         project = {'name': project}
 
@@ -226,6 +290,25 @@ def create_project(project, repository, **kwargs):
 def import_model_from_zip(name, project, file, description=None, version='latest'):
     # TODO: Allow import into folder if no project is given
     # TODO: Create new version if model already exists
+    """Import a model and contents as a ZIP file into a model project.
+
+    Parameters
+    ----------
+    name : str or dict
+        The name of the model.
+    project : str or dict
+        The name or id of the model project, or a dictionary representation of the project.
+    description : str
+        The description of the model.
+    file : byte
+        The ZIP file containing the model and contents.
+
+    Returns
+    -------
+    ResponseObj
+        The API response after importing the model.
+
+    """
     project = get_project(project)
 
     if project is None:
@@ -244,5 +327,3 @@ def import_model_from_zip(name, project, file, description=None, version='latest
                 headers={'Content-Type': 'application/octet-stream'})
 
     return r
-
-

tests/unit/test_model_management.py

@@ -62,7 +62,7 @@ def test_create_performance_definition():
             assert PROJECT['id'] == data['json']['projectId']
             assert MODEL['id'] in data['json']['modelIds']
             assert 'TestLibrary' == data['json']['dataLibrary']
-            assert 'TestData' == data['json']['dataTable']
+            assert 'TestData' == data['json']['dataPrefix']
             assert 'cas-shared-default' == data['json']['casServerId']
             assert data['json']['name'] is not None
             assert data['json']['description'] is not None