From 901bd2489344c79240ca37421512e5cb0f5f3edc Mon Sep 17 00:00:00 2001
From: marcus-sa <marcus-sa@outlook.com>
Date: Thu, 7 Mar 2024 14:09:20 +0100
Subject: [PATCH] feat(injector): improve injector context set method api

---
 packages/injector/src/injector.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/injector/src/injector.ts b/packages/injector/src/injector.ts
index 16bf9e41c..bea096862 100644
--- a/packages/injector/src/injector.ts
+++ b/packages/injector/src/injector.ts
@@ -914,8 +914,12 @@ export class InjectorContext {
         return this.getInjector(module || this.rootModule).instantiationCount(token, this.scope ? this.scope.name : scope);
     }
 
-    set(token: ContainerToken, value: any, module?: InjectorModule): void {
-        return this.getInjector(module || this.rootModule).set(token, value, this.scope);
+    set<T>(token: T, value: any, module?: InjectorModule): void {
+        return this.getInjector(module || this.rootModule).set(
+            getContainerToken(token),
+            value,
+            this.scope,
+        );
     }
 
     static forProviders(providers: ProviderWithScope[]) {