Skip to content

Commit c3136fe

Browse files
committed
Adjust ContextVarNode specialization
1 parent 5b082bc commit c3136fe

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/ContextvarsModuleBuiltins.java

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
/*
2-
* Copyright (c) 2019, 2023, Oracle and/or its affiliates. All rights reserved.
2+
* Copyright (c) 2019, 2024, Oracle and/or its affiliates. All rights reserved.
33
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
44
*
55
* The Universal Permissive License (UPL), Version 1.0
@@ -45,6 +45,7 @@
4545

4646
import java.util.List;
4747

48+
import com.oracle.graal.python.annotations.ArgumentClinic;
4849
import com.oracle.graal.python.builtins.Builtin;
4950
import com.oracle.graal.python.builtins.CoreFunctions;
5051
import com.oracle.graal.python.builtins.PythonBuiltinClassType;
@@ -54,8 +55,9 @@
5455
import com.oracle.graal.python.nodes.PRaiseNode;
5556
import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
5657
import com.oracle.graal.python.nodes.function.PythonBuiltinNode;
57-
import com.oracle.graal.python.nodes.function.builtins.PythonTernaryBuiltinNode;
58+
import com.oracle.graal.python.nodes.function.builtins.PythonTernaryClinicBuiltinNode;
5859
import com.oracle.graal.python.nodes.function.builtins.PythonUnaryBuiltinNode;
60+
import com.oracle.graal.python.nodes.function.builtins.clinic.ArgumentClinicProvider;
5961
import com.oracle.graal.python.runtime.PythonContext;
6062
import com.oracle.graal.python.runtime.object.PythonObjectFactory;
6163
import com.oracle.truffle.api.dsl.Bind;
@@ -87,8 +89,14 @@ protected Object copyCtx(
8789
}
8890

8991
@Builtin(name = "ContextVar", minNumOfPositionalArgs = 2, parameterNames = {"cls", "name", "default"}, constructsClass = PythonBuiltinClassType.ContextVar)
92+
@ArgumentClinic(name = "name", conversion = ArgumentClinic.ClinicConversion.TString)
9093
@GenerateNodeFactory
91-
public abstract static class ContextVarNode extends PythonTernaryBuiltinNode {
94+
public abstract static class ContextVarNode extends PythonTernaryClinicBuiltinNode {
95+
96+
@Override
97+
protected ArgumentClinicProvider getArgumentClinic() {
98+
return ContextvarsModuleBuiltinsClinicProviders.ContextVarNodeClinicProviderGen.INSTANCE;
99+
}
92100

93101
@Specialization
94102
protected static Object constructDef(@SuppressWarnings("unused") Object cls, TruffleString name, Object def,

0 commit comments

Comments
 (0)