Non-manual for OCaml

Disclaimer: this text is a draft for a potential new chapter for the OCaml reference manual. It will contain errors, approximations or confusing wordings. Proceed with caution.

Chapter 5  Polymorphism and its limitations

This chapter covers more advanced questions related to the limitations of polymorphic functions and types. There are some situations in OCaml where the type inferred by the type system may be less generic than expected. Such non-genericity can stem from interactions between side-effect and typing or the difficulties of implicit polymorphic recursion or higher-rank polymorphism.

This chapter details each of these situations and, if it is possible, how to recover genericity.

5.1  Weak polymorphism and mutation

5.1.1  Weakly polymorphic types

Maybe the most frequent examples of non-genericity derive from the interactions between polymorphic types and mutation. A simple example appears when typing the following expression

# let store = ref None ;;
 val store : '_a option ref = {contents = None}

Since the type of None is 'a option and the function ref has type 'b -> 'b ref, a natural deduction for the type of store would be 'a option ref. However, the inferred type, '_a option ref, is slightly different. Type variables whose name starts with an _ like '_a are weakly polymorphic type variables, sometimes shortened as weak type variables. A weak type variable is a placeholder for a single type that is currently unknown. Once the specific type t behind the placeholder type '_a known, all occurrences of '_a will be replaced by t. For instance, we can define another option reference and store an int inside:

# let another_store = ref None ;;
 val another_store : '_a option ref = {contents = None}
 
# another_store := Some 0;
another_store ;;
 - : int option ref = {contents = Some 0}

After storing an int inside another_store, the type of another_store has been updated from '_a option ref to int option ref. This distinction between weakly and generic polymorphic type variable protects OCaml programs from unsoundness and runtime errors. To understand from where unsoundness might come, consider this simple function which swaps a value x with the value stored inside a store reference, if there is such value:

# let swap store x = match !store with
  | None -> store := Some x; x
  | Some y -> store := Some x; y;;
 val swap : 'a option ref -> 'a -> 'a = <fun>

We can apply this function to our store

# let one = swap store 1
let one_again = swap store 2
let two = swap store 3;;
 val one : int = 1
val one_again : int = 1
val two : int = 2

After these three swaps the stored value is 3. Everything is fine up to now. We can then try to swap 2 with a more interesting value, for instance a function:

# let error = swap store (fun x -> x);;
 Error: This expression should not be a function, the expected type is
int

At this point, the type checker rightfully complains that it is not possible to swap an integer and a function, and that an int should always be traded for another int. Furthermore, the type checker prevents us to change manually the type of the value stored by store:

# store := Some (fun x -> x);;
 Error: This expression should not be a function, the expected type is
int

Indeed, looking at the type of store, we see that the weak type '_a has been replaced by the type int

# store;;
 - : int option ref = {contents = Some 3}

Therefore, after placing an int in store, we cannot use it to store any value other than an int. More generally, weak types protect the program from undue mutation of values with a polymorphic type.

Moreover, weak types cannot appear in the signature of toplevel modules: types must be known at compilation time. Otherwise, different compilation units could replace the weak type with different and incompatible types. For this reason, compiling the following small piece of code

let option_ref = ref None

yields a compilation error

Error: The type of this expression, '_a option ref,
       contains type variables that cannot be generalized

To solve this error, it suffices to add an explicit type annotation to specify the type at declaration time:

let option_ref: int option ref = ref None

This is in any case a good practice for such global mutable variables. Otherwise, they will pick out the type of first use. If there is a mistake at this point, this can result in confusing type errors when later, correct uses are flagged as errors.

5.1.2  The value restriction

Identifying the exact context in which polymorphic types should be replaced by weak types in a modular way is a difficult question. Indeed the type system must handle the possibility that functions may hide persistent mutable state. For instance, the following function uses an internal reference to implement a delayed identity function

# let make_fake_id () =
  let store = ref None in
  fun x -> swap store x ;;
 val make_fake_id : unit -> 'a -> 'a = <fun>
 
# let fake_id = make_fake_id();;
 val fake_id : '_a -> '_a = <fun>

It would be unsound to apply this fake_id function to values with different types. The function fake_id is therefore rightfully assigned the type '_a -> '_a rather than 'a -> 'a. At the same time, it ought to be possible to use a local mutable state without impacting the type of a function.

To circumvent these dual difficulties, the type checker considers that any value returned by a function might rely on persistent mutable states behind the scene and should be given a weak type. This restriction on the type of mutable values and the results of function application is called the value restriction. Note that this value restriction is conservative: there are situations where the value restriction is too cautious and gives a weak type to a value that could be safely generalized to a polymorphic type:

# let not_id = (fun x -> x) (fun x -> x);;
 val not_id : '_a -> '_a = <fun>

Quite often, this happens when defining function using higher order function. To avoid this problem, a solution is to add an explicit argument to the function:

# let id_again = fun x -> (fun x -> x) (fun x -> x) x;;
 val id_again : 'a -> 'a = <fun>

With this argument, id_again is seen as a function definition by the type checker and can therefore be generalized. This kind of manipulation is called eta-expansion in lambda calculus and is sometimes referred under this name.

5.1.3  The relaxed value restriction

There is another partial solution to the problem of unnecessary weak type, which is implemented directly within the type checker. Briefly, it is possible to prove that weak types that only appear as type parameters in covariant positions –also called positive positions– can be safely generalized to polymorphic types. For instance, the type 'a list is covariant in 'a:

#   let f () = [];;
 val f : unit -> 'a list = <fun>
 
#   let empty = f ();;
 val empty : 'a list = []

Remark that the type inferred for empty is 'a list and not '_a list that should have occurred with the value restriction since f () is a function application.

The value restriction combined with this generalization for covariant type parameters is called the relaxed value restriction.

5.1.4  Variance and value restriction

Variance describes how type constructors behave with respect to subtyping. Consider for instance a pair of type x and xy with xy a subtype of x, i.e xy :> x:

#   type x = [ `X ];;
 type x = [ `X ]
 
#   type xy = [ `X | `Y ];;
 type xy = [ `X | `Y ]

As x is a subtype of xy, we can convert a value of type x to a value of type xy:

#   let x:x = `X;;
 val x : x = `X
 
#   let x' = ( x :> xy);;
 val x' : xy = `X

Similarly, if we have a value of type x list, we can convert it to a value of type xy list, since we could convert each element one by one:

#   let l:x list = [`X; `X];;
 val l : x list = [`X; `X]
 
#   let l' = ( l :> xy list);;
 val l' : xy list = [`X; `X]

In other word, x :> xy implies that x list :> xy list, therefore the type constructor 'a list is covariant (it preserves subtyping) in its parameter 'a.

Contrarily, if we have a function that can handle values of type xy

#   let f = function
  | `X -> ()
  | `Y -> ();;
 val f : [< `X | `Y ] -> unit = <fun>

it can also handle values of type x:

#   let f' = (f :> x -> unit);;
 val f' : x -> unit = <fun>

Note that we can rewrite the type of f and f' as

#   type 'a proc = 'a -> unit
  let f' = (f: xy proc :> x proc);;
 type 'a proc = 'a -> unit
val f' : x proc = <fun>

In this case, we have x :> xy implies xy proc :> x proc. Notice that the second subtyping relation reverse the order of x and xy: the type constructor 'a proc is contravariant in its parameter 'a. More generally, the function type constructor 'a -> 'b is covariant in its return type 'b and contravariant in its argument type 'a.

A type constructor can also be invariant in some of its type parameters, neither covariant nor contravariant. A typical example is a reference:

#   let x: x ref = ref `X;;
 val x : x ref = {contents = `X}

If we were able to coerce x to the type xy ref as a variable xy, we could use xy to store the value `Y inside the reference and then use the x value to read this content as a value of type x, which would break the type system.

More generally, as soon as a type variable appears in a position describing mutable state it becomes invariant. As a corollary, covariant variables will never denote mutable locations and can be safely generalized. For a better description, interested reader can consult the original article by Jacque Guarrigue on http://www.math.nagoya-u.ac.jp/~garrigue/papers/morepoly-long.pdf

Together, the relaxed value restriction and type parameter covariance help to avoid eta-expansion in many situations.

5.1.5  Abstract data types

Moreover, when the type definitions are exposed, the type checker is able to infer variance information on its own and one can benefit from the relaxed value restriction even unknowingly. However, this is not the case anymore when defining new abstract types. As an illustration, we can define a module type collection as:

# module type COLLECTION = sig
  type 'a t
  val empty: unit -> 'a t
end

module Implementation = struct
  type 'a t = 'a list
  let empty ()= []
end;;
 module type COLLECTION = sig type 'a t val empty : unit -> 'a t end
module Implementation :
  sig type 'a t = 'a list val empty : unit -> 'a list end
 
# module List2: COLLECTION = Implementation;;
 module List2 : COLLECTION

In this situation, when coercing the module List2 to the module type COLLECTION, the type checker forgets that 'a List2.t was covariant in 'a. Consequently, the relaxed value restriction does not apply anymore:

#   List2.empty ();;
 - : '_a List2.t = <abstr>

To keep the relaxed value restriction, we need to declare the abstract type 'a COLLECTION.t as covariant in 'a:

# module type COLLECTION = sig
  type +'a t
  val empty: unit -> 'a t
end

module List2: COLLECTION = Implementation;;
 module type COLLECTION = sig type +'a t val empty : unit -> 'a t end
module List2 : COLLECTION

We then recover polymorphism:

#   List2.empty ();;
 - : 'a List2.t = <abstr>

5.2  Polymorphic recursion

The second major class of non-genericity is directly related to the problem of type inference for polymorphic function. In some circumstances, the type inferred by OCaml might be not general enough to allow the definition of some recursive functions on non-regular algebraic data type.

With a regular polymorphic algebraic data type, the type parameters of the type constructor are constant within the definition of the type. For instance, we can look at arbitrarily nested list defined as:

#   type 'a regular_nested = List of 'a list | Nested of 'a regular_nested list
  let l = Nested[ List [1]; Nested [List[2;3]]; Nested[Nested[]] ];;
 type 'a regular_nested = List of 'a list | Nested of 'a regular_nested list
val l : int regular_nested =
  Nested [List [1]; Nested [List [2; 3]]; Nested [Nested []]]

Note that the type constructor regular_nested always appears as 'a regular_nested in the definition above, with the same parameter 'a. Equipped with this type, one can compute a maximal depth with a classic recursive function

#   let rec maximal_depth = function
  | List _ -> 1
  | Nested [] -> 0
  | Nested (a::q) -> 1 + max (maximal_depth a) (maximal_depth (Nested q));;
 val maximal_depth : 'a regular_nested -> int = <fun>

Non-regular recursive algebraic data types correspond to polymorphic algebraic data types whose parameters types varies between the left and right side of the type definition. For instance, it might be interesting to define a datatype that ensures that all lists are nested at the same depth:

#   type 'a nested = List of 'a list | Nested of 'a list nested;;
 type 'a nested = List of 'a list | Nested of 'a list nested

Intuitively, a value of type 'a nested is a list of list …of list of elements a with k nested list. We can then adapt the maximal_depth function defined on regular_depth into a depth function that computes this k. As a first try, we may define

# let rec depth = function
  | List _ -> 1
  | Nested n -> 1 + depth n;;
 Error: This expression has type 'a list nested
       but an expression was expected of type 'a nested
       The type variable 'a occurs inside 'a list

The type error here comes from the fact that during the definition of depth, the type checker first assigns to it the type 'a -> 'b . When typing the pattern matching, it becomes 'a nested -> 'b then 'a nested -> int once the List branch is typed. However, when typing the application depth n in the Nested branch, the type checker encounters a problem: depth n is applied to 'a list nested, it must therefore have the type 'a list nested -> 'b. Unifying this constraint with the previous one leads to the impossible constraint 'a list nested = 'a nested. In other words, within its definition, the recursive function depth is applied to values of type 'a t with different types 'a due to the non-regularity of the type constructor nested. This creates a problem because the type checker had introduced a new type variable 'a only at the definition of the function depth whereas, here, it should have used a different type variable for every application of the function depth.

5.2.1  Explicitly polymorphic annotations

The solution of this conundrum is to use an explicitly polymorphic type annotation for the type 'a:

# let rec depth: 'a. 'a nested -> int = function
  | List _ -> 1
  | Nested n -> 1 + depth n;;
 val depth : 'a nested -> int = <fun>
 
# depth ( Nested(List [ [7]; [8] ]) );;
 - : int = 2

In the type of depth, 'a.'a nested -> int, the type variable 'a is universally quantified, i.e 'a.'a nested -> int reads as “for all type 'a, depth maps 'a nested values to integers”. Whereas the standard type 'a nested -> int can be interpreted as “let be a type variable 'a, then depth maps 'a nested values to integers”. There is two major differences with these two type expressions. First, the explicit polymorphic annotation indicates to the type checker that it needs to introduce a new type variable every times the function depth is applied. This solves our problem with the definition of the function depth.

Second, it also notifies the type checker that the type of the function should be polymorphic. Indeed, without explicit polymorphic type annotation, the following type annotation is perfectly valid

#   let sum: 'a -> 'b -> 'c = fun x y -> x + y;;
 val sum : int -> int -> int = <fun>

since 'a,'b and 'c denote type variables that may or may not be polymorphic. Whereas, it is an error to unify an explicitly polymorphic type with a non-polymorphic type:

#   let sum: 'a 'b 'c. 'a -> 'b -> 'c = fun x y -> x + y;;
 Error: This definition has type int -> int -> int which is less general than
         'a 'b 'c. 'a -> 'b -> 'c

An important remark here is that it is not needed to explicit fully the type of depth: it is sufficient to add annotations only for the universally quantified type variables:

# let rec depth: 'a. 'a nested -> _ = function
  | List _ -> 1
  | Nested n -> 1 + depth n;;
 val depth : 'a nested -> int = <fun>
 
# depth ( Nested(List [ [7]; [8] ]) );;
 - : int = 2

5.2.2  More examples

With explicit polymorphic annotations, it becomes possible to implement any recursive function that depends only on the structure of the nested lists and not on the type of the elements. For instance, a more complex example would be to compute the total number of elements of the nested lists:

#   let len nested =
    let map_and_sum f = List.fold_left (fun acc x -> acc + f x) 0 in
    let rec len: 'a. ('a list -> int ) -> 'a nested -> int =
    fun nested_len n ->
      match n with
      | List l -> nested_len l
      | Nested n -> len (map_and_sum nested_len) n
    in
  len List.length nested;;
 val len : 'a nested -> int = <fun>
 
# len (Nested(Nested(List [ [ [1;2]; [3] ]; [ []; [4]; [5;6;7]]; [[]] ])));;
 - : int = 7

Similarly, it may be necessary to use more than one explicitly polymorphic type variables, like for computing the nested list of list lengths of the nested list:

# let shape n =
  let rec shape: 'a 'b. ('a nested -> int nested) ->
    ('b list list -> 'a list) -> 'b nested -> int nested
    = fun nest nested_shape ->
      function
      | List l -> raise
       (Invalid_argument "shape requires nested_list of depth greater than 1")
      | Nested (List l) -> nest @@ List (nested_shape l)
      | Nested n ->
        let nested_shape = List.map nested_shape in
        let nest x = nest (Nested x) in
        shape nest nested_shape n in
  shape (fun n -> n ) (fun l -> List.map List.length l ) n;;
 val shape : 'a nested -> int nested = <fun>
 
# shape (Nested(Nested(List [ [ [1;2]; [3] ]; [ []; [4]; [5;6;7]]; [[]] ])));;
 - : int nested = Nested (List [[2; 1]; [0; 1; 3]; [0]])

5.3  Higher-rank polymorphic functions

Explicit polymorphic annotations are however not sufficient to cover all the cases where the inferred type of a function is less general than expected. A similar problem arises when using polymorphic functions as arguments of a higher-order functions. For instance, we may want to compute the average depth or length of two nested lists:

#   let average_depth x y = (depth x + depth y) / 2;;
 val average_depth : 'a nested -> 'b nested -> int = <fun>
 
#   let average_len x y = (len x + len y) / 2;;
 val average_len : 'a nested -> 'b nested -> int = <fun>
 
#   let one = average_len (List [2]) (List [[]]);;
 val one : int = 1

It would be natural to factorize these two definitions as:

#     let average f x y = (f x + f y) / 2;;
 val average : ('a -> int) -> 'a -> 'a -> int = <fun>

However, the type of average len is less generic than the type of average_len, since it requires the type of the first and second argument to be the same:

#   average_len (List [2]) (List [[]]);;
 - : int = 1
 
#   average len (List [2]) (List [[]]);;
 Error: This expression has type 'a list
       but an expression was expected of type int

As previously with polymorphic recursion, the problem stems from the fact that type variables are introduced only at the start of the let definitions. When we compute both f x and f y, the type of x and y are unified together. To avoid this unification, we need to indicate to the type checker that f is polymorphic in its first argument. In some sense, we would want average to have type

val average: ('a. 'a nested -> int) -> 'a nested -> 'b nested -> int

Note that this syntax is not valid within OCaml: average has an universally quantified type 'a inside the type of one of its argument whereas for polymorphic recursion the universally quantified type was introduced before the rest of the type. This position of the universally quantified type means that average is a second-rank polymorphic function. This kind of higher-rank functions is not directly supported by OCaml: type inference for second-rank polymorphic function and beyond is undecidable; therefore using this kind of higher-rank functions requires to handle manually these universally quantified types.

In OCaml, there are two ways to introduce this kind of explicit universally quantified types: universally quantified record fields,

#   type 'a nested_reduction = { f:'elt. 'elt nested -> 'a };;
 type 'a nested_reduction = { f : 'elt. 'elt nested -> 'a; }
 
#   let boxed_len = { f = len };;
 val boxed_len : int nested_reduction = {f = <fun>}

and universally quantified object methods:

#   let obj_len = object method f:'a. 'a nested -> 'b = len end;;
 val obj_len : < f : 'a. 'a nested -> int > = <obj>

To solve our problem, we can therefore use either the record solution:

#   let average nsm x y = (nsm.f x + nsm.f y) / 2 ;;
 val average : int nested_reduction -> 'a nested -> 'b nested -> int = <fun>

or the object one:

#   let average (obj:<f:'a. 'a nested -> _ > ) x y = (obj#f x + obj#f y) / 2 ;;
 val average : < f : 'a. 'a nested -> int > -> 'b nested -> 'c nested -> int =
  <fun>