fix: Make d.Infer<T> assignable to d.InferInput<T> (#2335)

Author: iwoplaza

packages/typegpu/src/shared/repr.ts

@@ -38,7 +38,9 @@ export type Infer<T> = T extends { readonly [$repr]: infer TRepr } ? TRepr : T;
  * const arrayOfStructs = d.arrayOf(d.struct({ pos: d.vec3f, id: d.f32 }), 4);
  * type D = d.InferInput<typeof arrayOfStructs>; // { pos: d.v3f | Float32Array<ArrayBufferLike> | [number, number, number]; id: number; }[]
  */
-export type InferInput<T> = T extends { readonly [$inRepr]: infer TRepr } ? TRepr : Infer<T>;
+export type InferInput<T> =
+  | Infer<T>
+  | (T extends { readonly [$inRepr]: infer TRepr } ? TRepr : never);
 
 /**
  * Extracts a sparse/partial inferred representation of a resource.

packages/typegpu/tests/buffer.test.ts

@@ -8,7 +8,6 @@ import { getName } from '../src/shared/meta.ts';
 import type { IsValidBufferSchema, IsValidUniformSchema } from '../src/shared/repr.ts';
 import type { TypedArray } from '../src/shared/utilityTypes.ts';
 import { it } from 'typegpu-testing-utility';
-import { buildWriter } from '../src/data/compiledIO.ts';
 
 function toUint8Array(...arrays: Array<TypedArray>): Uint8Array {
   let totalByteLength = 0;
@@ -846,7 +845,7 @@ describe('TgpuBuffer (InferInput)', () => {
     expectTypeOf(arrBuf.write)
       .parameter(0)
       .toEqualTypeOf<
-        (d.v3f | [number, number, number] | Float32Array)[] | Float32Array | ArrayBuffer
+        (d.v3f | [number, number, number] | Float32Array)[] | Float32Array | ArrayBuffer | d.v3f[]
       >();
 
     expectTypeOf(scalarArrBuf.write)

packages/typegpu/tests/inferInput.test.ts

@@ -0,0 +1,13 @@
+import { test } from 'typegpu-testing-utility';
+import { describe, expect } from 'vitest';
+import { d, ValidateBufferSchema } from 'typegpu';
+
+describe('d.InferInput', () => {
+  test('d.Infer<T> should be assignable to d.InferInput<T>', ({ root }) => {
+    function foo<T extends d.AnyWgslData>(schema: ValidateBufferSchema<T>, input: d.Infer<T>) {
+      return root.createBuffer(schema, input);
+    }
+
+    expect(() => foo(d.f32, 1)).not.toThrow();
+  });
+});