t, ok := db.table(s.table)
		db.mu.Unlock()
		if !ok {
			return nil, fmt.Errorf("fakedb: table %q doesn't exist", s.table)
		}

		if s.table == "magicquery" {
			if len(s.whereCol) == 2 && s.whereCol[0].Column == "op" && s.whereCol[1].Column == "millis" {
				if args[0].Value == "sleep" {
					time.Sleep(time.Duration(args[1].Value.(int64)) * time.Millisecond)
				}
			}
		}

        TypeID::get<TF::TruncateModOp>(),
        TypeID::get<TF::UniqueOp>(),
        TypeID::get<TF::UnpackOp>(),
        TypeID::get<TF::UpperBoundOp>(),
        TypeID::get<TF::WhereOp>(),
        TypeID::get<TF::XlaSendTPUEmbeddingGradientsOp>(),
        TypeID::get<TF::XlaBroadcastHelperOp>(),
        TypeID::get<TF::XlaCallModuleOp>(),
        TypeID::get<TF::XlaCustomCallV2Op>(),

    bool allow_string_consts = true;

    // Whether to allow the compilation of CollectiveReduceV2Op.
    bool allow_collective_reduce_v2 = true;

    // Whether to allow the compilation of WhereOp.
    bool allow_where_op = true;

    // Whether to allow the compilation of UniqueOp. Compilation of the UniqueOp
    // generates output with bounded dynamic shape that may cause failures with
    // auto clustering.

      OperationName(mlir::TF::DynamicPartitionOp::getOperationName(), context),
      OperationName(mlir::TF::OutfeedEnqueueOp::getOperationName(), context),
      OperationName(mlir::TF::WhereOp::getOperationName(), context),
      OperationName(mlir::TF::UniqueOp::getOperationName(), context),
      OperationName(mlir::TF::XlaSetDynamicDimensionSizeOp::getOperationName(),
                    context),