fix: types

zazuko · Jan 8, 2025 · 96f620f · 96f620f
1 parent 779422a
commit 96f620f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/packages/env/lib/DatasetFactory.ts b/packages/env/lib/DatasetFactory.ts
@@ -5,6 +5,8 @@ import type DatasetCore from '@rdfjs/dataset/DatasetCore.js'
 import addAll from 'rdf-dataset-ext/addAll.js'
 import deleteMatch from 'rdf-dataset-ext/deleteMatch.js'
 import equals from 'rdf-dataset-ext/equals.js'
+import type { TermMapFactory } from '@rdfjs/term-map/Factory.js'
+import type DataFactory from '@rdfjs/data-model/Factory.js'
 import type { Dataset, DatasetCtor } from './Dataset.js'
 
 export interface FactoryMethod<D extends DatasetCore> {
@@ -19,10 +21,10 @@ export interface DatasetFactory<D extends DatasetCore = Dataset> {
   dataset: FactoryMethod<D>
 }
 
-export default <D extends DatasetCore>(createConstructor: (env: Environment<FormatsFactory>) => DatasetCtor<D>) => class implements DatasetFactory<D> {
+export default <D extends DatasetCore>(createConstructor: (env: Environment<FormatsFactory | DataFactory | TermMapFactory>) => DatasetCtor<D>) => class implements DatasetFactory<D> {
   public dataset!: FactoryMethod<D>
 
-  init(this: Environment<FormatsFactory | DatasetFactory<D>>) {
+  init(this: Environment<FormatsFactory | DatasetFactory<D> | DataFactory | TermMapFactory>) {
     const Dataset = createConstructor(this)
 
     this.dataset = ((quads: Iterable<Quad> = []) => {

diff --git a/packages/env/lib/DatasetFactoryExt.ts b/packages/env/lib/DatasetFactoryExt.ts
@@ -26,7 +26,7 @@ export interface DatasetFactoryExt<D extends DatasetCore = Dataset> {
   dataset: FactoryMethod<D>
 }
 
-export default <D extends DatasetCore>(createConstructor: (env: Environment<FormatsFactory>) => DatasetCtor<D>) => class extends DatasetFactory(createConstructor) implements DatasetFactoryExt<D> {
+export default <D extends DatasetCore>(createConstructor: (env: Environment<FormatsFactory | DataFactory | TermMapFactory>) => DatasetCtor<D>) => class extends DatasetFactory(createConstructor) implements DatasetFactoryExt<D> {
   public declare dataset: FactoryMethod<D>
 
   init(this: Environment<FormatsFactory | DatasetFactoryExt<D> | DataFactory | TermMapFactory>) {