fix(lib): fix TypeScript definition

azu · Aug 24, 2019 · ece436d · ece436d
1 parent a5f08a9
commit ece436d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/comment-to-assert.ts b/src/comment-to-assert.ts
@@ -77,7 +77,7 @@ export function toAssertFromSource(code: string, options?: toAssertFromSourceOpt
 /**
  * transform AST to asserted AST.
  */
-export function toAssertFromAST(ast: File, options: wrapAssertOptions = {}): File {
+export function toAssertFromAST<T extends File>(ast: T, options: wrapAssertOptions = {}): T {
     const replaceSet = new Set();
     let id = 0;
     traverse(ast, {