diff --git a/mlflow/server/js/src/model-registry/components/ModelVersionTable.test.tsx b/mlflow/server/js/src/model-registry/components/ModelVersionTable.test.tsx index fef434984fd73..95cbfb5208407 100644 --- a/mlflow/server/js/src/model-registry/components/ModelVersionTable.test.tsx +++ b/mlflow/server/js/src/model-registry/components/ModelVersionTable.test.tsx @@ -76,7 +76,7 @@ describe('ModelVersionTable', () => { modelName: modelName, usingNextModelsUI: true, modelVersions: [mockModelVersionDetailed(modelName, 1, Stages.NONE, ModelVersionStatus.READY)], - aliases: [{alias: 'champion', version: '1'}] + aliases: [{ alias: 'champion', version: '1' }], }; renderWithProviders(); expect(screen.queryByRole('columnheader', { name: 'Stage' })).not.toBeInTheDocument(); diff --git a/mlflow/server/js/src/model-registry/components/ModelVersionTable.tsx b/mlflow/server/js/src/model-registry/components/ModelVersionTable.tsx index 9a363b8f8bed8..0d54d68c3fe30 100644 --- a/mlflow/server/js/src/model-registry/components/ModelVersionTable.tsx +++ b/mlflow/server/js/src/model-registry/components/ModelVersionTable.tsx @@ -57,7 +57,7 @@ type ModelVersionTableProps = { modelEntity?: ModelEntity; onMetadataUpdated: () => void; usingNextModelsUI: boolean; - aliases?: ModelAliasMap + aliases?: ModelAliasMap; }; type ModelVersionColumnDef = ColumnDef & { @@ -83,7 +83,7 @@ export const ModelVersionTable = ({ modelEntity, onMetadataUpdated, usingNextModelsUI, - aliases + aliases, }: ModelVersionTableProps) => { const aliasesByVersion = useMemo(() => { const result: Record = {}; @@ -92,7 +92,7 @@ export const ModelVersionTable = ({ result[version] = []; } result[version].push(alias); - }) + }); return result; }, [aliases]); const versions = useMemo( diff --git a/mlflow/server/js/src/shared/web-shared/snippet/index.tsx b/mlflow/server/js/src/shared/web-shared/snippet/index.tsx index a412bb814028f..f7e77a0b453b8 100644 --- a/mlflow/server/js/src/shared/web-shared/snippet/index.tsx +++ b/mlflow/server/js/src/shared/web-shared/snippet/index.tsx @@ -1,7 +1,9 @@ import { PrismLight as SyntaxHighlighter } from 'react-syntax-highlighter'; import python from 'react-syntax-highlighter/dist/cjs/languages/prism/python'; +import json from 'react-syntax-highlighter/dist/cjs/languages/prism/json'; SyntaxHighlighter.registerLanguage('python', python); +SyntaxHighlighter.registerLanguage('json', json); import duotoneDarkStyle from './theme/databricks-duotone-dark'; import lightStyle from './theme/databricks-light'; @@ -22,7 +24,7 @@ const themesStyles: Record = { duotoneDark: duotoneDarkStyle, }; -export type CodeSnippetLanguage = 'python'; +export type CodeSnippetLanguage = 'python' | 'json'; export interface CodeSnippetProps { /**