Skip to content

Commit

Permalink
Add model management functions to io namespace in generated docs (ten…
Browse files Browse the repository at this point in the history
…sorflow#1082)

Add model management functions to io namespace in generated docs. Fixes tensorflow/tfjs#354

DOC
  • Loading branch information
tafsiri authored and dsmilkov committed Jun 6, 2018
1 parent e12aebd commit 295233e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/io/model_management.ts
Original file line number Diff line number Diff line change
Expand Up @@ -189,7 +189,7 @@ export class ModelManagement {
* 'indexeddb://my/model/1'. Model artifacts info include type of the
* model's topology, byte sizes of the topology, weights, etc.
*/
@doc({heading: 'Models', subheading: 'Management'})
@doc({heading: 'Models', subheading: 'Management', namespace: 'io'})
static async listModels(): Promise<{[url: string]: ModelArtifactsInfo}> {
const schemes = ModelStoreManagerRegistry.getSchemes();
const out: {[url: string]: ModelArtifactsInfo} = {};
Expand Down Expand Up @@ -230,7 +230,7 @@ export class ModelManagement {
* is successful).
* @throws Error if deletion fails, e.g., if no model exists at `path`.
*/
@doc({heading: 'Models', subheading: 'Management'})
@doc({heading: 'Models', subheading: 'Management', namespace: 'io'})
static async removeModel(url: string): Promise<ModelArtifactsInfo> {
const schemeAndPath = parseURL(url);
const manager = ModelStoreManagerRegistry.getManager(schemeAndPath.scheme);
Expand Down Expand Up @@ -277,7 +277,7 @@ export class ModelManagement {
* @throws Error if copying fails, e.g., if no model exists at `sourceURL`, or
* if `oldPath` and `newPath` are identical.
*/
@doc({heading: 'Models', subheading: 'Management'})
@doc({heading: 'Models', subheading: 'Management', namespace: 'io'})
static async copyModel(sourceURL: string, destURL: string):
Promise<ModelArtifactsInfo> {
const deleteSource = false;
Expand Down Expand Up @@ -323,7 +323,7 @@ export class ModelManagement {
* @throws Error if moving fails, e.g., if no model exists at `sourceURL`, or
* if `oldPath` and `newPath` are identical.
*/
@doc({heading: 'Models', subheading: 'Management'})
@doc({heading: 'Models', subheading: 'Management', namespace: 'io'})
static async moveModel(sourceURL: string, destURL: string):
Promise<ModelArtifactsInfo> {
const deleteSource = true;
Expand Down

0 comments on commit 295233e

Please sign in to comment.